fenglf / Kaggle-dog-breed-classification

This is the baseline of Kaggle-dog-breed-classification on Python, Keras, and TensorFlow.
30 stars 12 forks source link

Could you share the weight file which in(/home/fenglf/PycharmProjects/keras-finetuning-master/model/pretrained/) #2

Open KeuntekLee opened 6 years ago

KeuntekLee commented 6 years ago

there are some weight files need to run code /home/fenglf/PycharmProjects/keras-finetuning-master/model/pretrained/ please share h5 file in above path

fenglf commented 6 years ago

it just comes from imagenet. replace "base_model_name(weights=pretrained_model_dir, include_top=False)" with "base_model_name(weights='imagenet', include_top=False)"
should solve.