API обнаружения объектов Tensorflow не обнаруживает ограничивающие рамки - PullRequest
0 голосов
/ 24 декабря 2018

Мой вопрос похож на Tensorflow Object Detection API не обнаруживает объекты правильно / не обнаруживает объекты вообще , но это другой вариант использования, и этот вопрос не решен.

Я использую API обнаружения объектов для обучения передачи, чтобы идентифицировать лица (1 класс).Используется предварительно обученная модель ssd_mobilenet_v1_coco_11_06_2017.Это было выполнено в течение 1,5 часов и для 3500 итераций, и потери составили около 4,5 (первые несколько итераций резко упали, как и ожидалось, поэтому я предполагаю, что это не лучшая модель, но я должен получить заметные результаты)

Затем я использовал последнюю контрольную точку, экспортированный граф вывода, внес изменения в записную книжку object_detection_tutorial.Однако это просто выводит мои тестовые изображения без каких-либо ограничивающих рамок (ПРИМЕЧАНИЕ: - оригинальный блокнот работает нормально, и я могу видеть рамки, поэтому я предполагаю, что есть некоторая проблема с моей моделью, которую я не могу выяснить)

Я вставляю результаты, которые я использовал для отладки.Похоже, координаты ограничивающего прямоугольника имеют 1 и 0.Я также поиграл, изменив min_score_thresh в visualization_utils.py, но это тоже не помогло.

Фрагменты файла конфигурации (внесены изменения в образец. Изменены-num_classes на 1 и пути к моим относящимся файлам)

model {
  ssd {
    num_classes: 1
    box_coder {
      faster_rcnn_box_coder {
        y_scale: 10.0
        x_scale: 10.0
        height_scale: 5.0
        width_scale: 5.0
      }

fine_tune_checkpoint: "ssd_mobilenet_v1_coco_11_06_2017/model.ckpt"

train_input_reader: {
  tf_record_input_reader {
    input_path: "training/train.record"
  }
  label_map_path: "training/face-detection.pbtxt"
}

Пример выходных данных Detection_boxes

array([[0.        , 0.2554323 , 1.        , 0.7534608 ],
       [0.3012189 , 0.        , 0.70924723, 1.        ],
       [0.18439636, 0.        , 0.817055  , 1.        ],
       [0.05466348, 0.06271043, 0.95918256, 0.94578576],
       [0.5260097 , 0.07892245, 0.97103214, 1.        ],
       [0.27985883, 0.5130247 , 1.        , 1.        ],
       [0.        , 0.41663095, 0.5991725 , 1.        ],
       [0.03816789, 0.        , 0.4441032 , 1.        ],
       [0.        , 0.5875985 , 1.        , 0.9825353 ],
       [0.5518031 , 0.        , 0.9699122 , 0.9448744 ],
       [0.        , 0.        , 0.6656574 , 0.7552371 ],
       [0.        , 0.07883103, 1.        , 0.49962282],
       [0.04292154, 0.57037276, 1.        , 0.9573701 ],
       [0.05089287, 0.00120801, 0.46899152, 1.        ],
       [0.43095234, 0.31733978, 1.        , 1.        ],

Вывод обнаружения_класса

[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
   1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
   1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
   1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
   1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]

выборка вывода детектирования_отчета низкие оценки, но изменение порога не помогает

   3.20677400e-01, 3.19893241e-01, 3.14980447e-01, 3.09631795e-01,
   2.53045678e-01, 2.52637416e-01, 2.52323598e-01, 2.41894022e-01,
   2.40268022e-01, 2.40083978e-01, 2.36307651e-01, 2.15435311e-01,
   2.14362189e-01, 2.13490084e-01, 2.12984398e-01, 2.10590839e-01,
   2.03191489e-01, 1.66917413e-01, 1.64012805e-01, 1.60537049e-01,
   6.71069846e-02, 6.55404702e-02, 3.69835161e-02, 3.63573395e-02,
   3.15450728e-02, 2.46258732e-02, 2.34935526e-02, 2.04152763e-02,
   1.82382613e-02, 1.56786684e-02, 1.50712561e-02, 1.47904372e-02,
   1.45556601e-02, 1.41305756e-02, 1.14051756e-02, 1.02278581e-02,
   9.46425926e-03, 8.77886359e-03, 8.67216382e-03, 7.98975583e-03
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...