titu1994 / DenseNet

DenseNet implementation in Keras
MIT License
707 stars 294 forks source link

'global_average_pooling2d_1' doesn't exist in 'DenseNet-BC-121-32-no-top.h5' #24

Closed Mudwall closed 7 years ago

Mudwall commented 7 years ago

I try "model = DenseNetImageNet121((224,224,3),classes=5, weights='imagenet' ,include_top=False)" and I have downloaded the 'DenseNet-BC-121-32-no-top.h5', so 'model.load_weights( 'DenseNet-BC-121-32-no-top.h5')' is executed. But the error raised as blowe : KeyError: "Unable to open object (Object 'global_average_pooling2d_1' doesn't exist)" Anaconda3,pythin3.6,in Win7.

titu1994 commented 7 years ago

Oh that's a slight bug. Ill take care of it in some time.

titu1994 commented 7 years ago

Fixed via 003c562cfd1bc4856cc00c23590705544dd65b74