50-layer Residual Network - ImageNet

see https://github.com/keras-team/keras/blob/master/keras/applications/resnet50.py



In [1]:

    
KERAS_MODEL_FILEPATH = '../../demos/data/resnet50/resnet50.h5'



In [2]:

    
from keras.applications import resnet50, imagenet_utils









    



Using TensorFlow backend.
/home/leon/miniconda3/lib/python3.6/importlib/_bootstrap.py:219: RuntimeWarning: compiletime version 3.5 of module 'tensorflow.python.framework.fast_tensor_util' does not match runtime version 3.6
  return f(*args, **kwds)



In [3]:

    
input_shape = (224, 224, 3)

model = resnet50.ResNet50(input_shape=input_shape, include_top=True, weights='imagenet')



In [4]:

    
model.save(KERAS_MODEL_FILEPATH)



In [ ]: