Tensorflow Модель SSD Mobilenet Пороговое значение обнаружения Править - PullRequest
Новая
       2

Tensorflow Модель SSD Mobilenet Пороговое значение обнаружения Править

0 голосов
/ 17 октября 2019

Я пытаюсь обучить SSD_MobileNet V1 в пользовательском наборе данных для обнаружения мясного скота на ферме. Для обучения модели я следовал инструкциям по senddex . Это работает прямо сейчас. Однако в выходном изображении он рисует прямоугольники только вокруг объектов со значением достоверности обнаружения, превышающим 50%. Ни один обнаруженный объект на выходном изображении не получил достоверность обнаружения ниже 50%. Я пытаюсь установить пороговое значение на 0,3.

Для этого я отредактировал файл ssd_mobilenet_v1_pets.config из - 

argmax_matcher {
    matched_threshold: 0.5
    unmatched_threshold: 0.5
    ignore_thresholds: false
    negatives_lower_than_unmatched: true
    force_match_for_each_row: true
  }

в 

argmax_matcher {
    matched_threshold: 0.3
    unmatched_threshold: 0.3
    ignore_thresholds: false
    negatives_lower_than_unmatched: true
    force_match_for_each_row: true
  }

Затем экспортировал граф вывода и использовал этот граф. Однако на выходе ничего не изменилось, каждый обнаруженный объект получил более 50% достоверности. Есть ли простой способ изменить пороговое значение?

...