можно пометить собаку этой моделью? - PullRequest
0 голосов
/ 29 сентября 2019

Я видел этот код на 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')

после того, как он сохранит его, он может использовать его на некоторой картинке и получить ответ для записи

Я хочу добавить к нему возможность пометить собакуили кат.

1 Ответ

0 голосов
/ 30 сентября 2019

Вопросы не ясны, но, позвольте мне ответить по одному.

1) В: Могу ли я использовать этот код и отметить собаку / кошку на картинке?

A: Да, вы можете.Тем не менее, вы должны иметь сохраненную модель, чтобы использовать ее.В противном случае вы должны обучить модель и сохранить ее снова.

2) В: Я хочу добавить к ней возможность маркировки собаки иликошка.Возможно ли это?

* A: Да, это возможно.model.predict даст вам вероятности.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...