Обнаружение объектов одного типа с использованием чистых керас - PullRequest
Новая
       16

Обнаружение объектов одного типа с использованием чистых керас

0 голосов
/ 23 марта 2020

Как организовать работу двоичного классификатора и детектора? Они должны быть отдельными (первый вариант)
или они могут иметь общие слои (второй вариант)?

Первый вариант: 

base_model_1 = tf.keras.applications.SOMEMODEL(...)
classificator = tf.keras.Sequential([
  base_model_1,
  # ...
  tf.keras.layers.Dense(1,activation='sigmoid')
])

base_model_2 = tf.keras.applications.SOMEMODEL(...)
detector = tf.keras.Sequential([
  base_model_2,
  # ...
  tf.keras.layers.Dense(4,activation='sigmoid')
])

Второй вариант: 

base_model = tf.keras.applications.SOMEMODEL(...)
model = tf.keras.Sequential([
  base_model,
  # ...
  tf.keras.layers.Dense(5,activation='sigmoid')
])

Я хочу сделать второй вариант. Но я не знаю входные координаты bbox для фоновых (это не объект) картинок, которые нужны для обучения.

...