keras官方提供的model

初始化狀態可以設置weights可以為imagenet或None

以及include_top可以為True或False

 

====VGG16 model====

from keras.applications.vgg16 import VGG16

model = VGG16(weights='imagenet', include_top=True)

 

====VGG19 model====

from keras.applications.vgg19 import VGG19

model = VGG19(weights='imagenet', include_top=True)

 

====Resnet50====

from keras.applications.resnet50 import ResNet50

model = ResNet50(weights='imagenet', include_top=True)

 

arrow
arrow

    楓綺 發表在 痞客邦 留言(0) 人氣()