Предсказать только ограничивающую рамку

Question

У меня две папки для обучения и тестирования.Каждая папка имеет 14k изображений случайного одного объекта, такого как стул, ящик, веер, банка и т. Д. В дополнение к этому у меня есть 4 столбца [x1, x2, y1, y2] для каждого изображения обучающего наборав котором заключен этот случайный объект (ограничивающий прямоугольник).С помощью этой информации я хочу предсказать ограничивающие рамки для тестового набора.
Я очень новичок в Computer Vision. Было бы очень полезно, если бы кто-нибудь мог помочь мне, как начать с обучения такого рода моделей.Я нашел yolov3, но он также включает классификацию.

Answer 1

Я рекомендую вам увидеть код github здесь .

В обнаружении есть функция do_detect () .

эта функциявозвращает класс и bbox, которые вы хотите получить из изображения.

boxes = do_detect(model, image, confidence_threshold, nms_threshold)