Я видел этот код на YouTube Как использовать вашу обученную модель - Основы глубокого обучения с Python, TensorFlow и Keras с.6
Если я принесу фотографию, он скажет мнеесли это изображение кошки или собаки.
Я могу использовать этот код и пометить собаку / кошку на рисунке?
Или мне нужно запустить совершенно новую модель?
в коде я получаю все изображения собаки и кошки и помещаю их в X и y
X = это особенности
y = это метка
код
dense_layers = [0]
layer_sizes = [64]
conv_layers = [3]
model = Sequential()
model.add(Conv2D(layer_size, (3, 3), input_shape=X.shape[1:]))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(layer_size, (3, 3)))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Flatten())
model.add(Dense(layer_size))
model.add(Activation('relu'))
model.add(Dense(1))
model.add(Activation('sigmoid'))
model.compile(loss='binary_crossentropy',
optimizer='adam',
metrics=['accuracy'],
)
model.fit(X, y,
batch_size=32,
epochs=10,
validation_split=0.3,
)
model.save('64x3-CNN.model')
после того, как он сохранит его, он может использовать его на некоторой картинке и получить ответ для записи
Я хочу добавить к нему возможность пометить собакуили кат.