Darkflow: Использование darkflow из другого приложения Python для сохранения изображения с прогнозами и ограничивающими рамками

Вам нужно будет написать свою собственную функцию для рисования ящиков и записи изображения после анализа вывода json.

Вы можете использовать такой код

cv2.rectangle(imgcv,
              (result["topleft"]["x"], result["topleft"]["y"]),
              (result["bottomright"]["x"],
               result["bottomright"]["y"]),
              (0, 255, 0), 4)
text_x, text_y = result["topleft"][
    "x"] - 10, result["topleft"]["y"] - 10

cv2.putText(imgcv, result["label"], (text_x, text_y),
            cv2.FONT_HERSHEY_SIMPLEX, 0.8, (0, 255, 0), 2, cv2.LINE_AA)

Где результат — это один элемент из списка результатов, возвращаемых методом прогнозирования возврата, а imgcv — это исходное изображение. Надеюсь, это поможет!

Спасибо !

Можно ли как-то сохранить изображение с прогнозами и их ограничительными рамками после печати результата?

из darkflow.net.build импортировать TFNet
импорт cv2

options = {"model": "cfg/yolo.cfg", "load": "bin/yolo.weights", "threshold": 0,1}

tfnet = TFNet (опции)

imgcv = cv2.imread("./sample_img/sample_dog.jpg")
результат = tfnet.return_predict(imgcv)
печать (результат)

У меня возникла проблема при выполнении этого кода. Не могли бы вы рассказать мне о параметре «загрузка».

Можно ли как-то сохранить изображение с прогнозами и их ограничительными рамками после печати результата?
из darkflow.net.build импортировать TFNet
импорт cv2
options = {"model": "cfg/yolo.cfg", "load": "bin/yolo.weights", "threshold": 0,1}
tfnet = TFNet (опции)
imgcv = cv2.imread("./sample_img/sample_dog.jpg")
результат = tfnet.return_predict(imgcv)
печать (результат)

У меня возникла проблема при выполнении этого кода. Не могли бы вы рассказать мне о параметре «загрузка».

@abhishek795jha Параметр «загрузка» - это ваши файлы веса ....... изначально мы берем файл веса yolo, после тренировки вы создаете файл .pb для тестирования в качестве файла веса. Во время обучения вы должны использовать команду «--savepb», чтобы сохранить файл .pb с помощью команды потока.