def create_resnet50(input_img):
net = ResNet50(weights='imagenet', include_top=False,
input_tensor=input_img)
for layer in net.layers[1:]:
layer.trainable = False
net = Reshape((-1,))(net.outputs[0])
return net
评论列表
文章目录