Мой вопрос похож на 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