版权声明:本文为博主原创文章,转载请注明出处。 https://blog.csdn.net/nima1994/article/details/82686132
代码:
import keras
keras.utils.plot_model(keras.applications.ResNet50(include_top=True,input_shape=(224,224,3),weights=None), to_file='image_model.png', show_shapes=True)
ResNet50的标准输入为224x224,avg_pool
(-3层)及之后为 include_top=False/True
多出的。