close

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
    創作者介紹
    創作者 楓綺 的頭像
    楓綺

    K_程式人

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