Я слежу за этим GitHub repo по реализации yolo v3 на изображениях / видео / камерах.
Я начал с попытки запустить его на изображении через командную строку, и похоже, что он работает (он печатает обнаруженный объект).
Однако модель, похоже, не сохраняет новое изображение (с обнаруженным объектом и ограничивающими рамками) в виде нового файла изображения.
Я считаю, что часть, где изображение должно быть сохранено, -
list(map(lambda x: write(x, im_batches, orig_ims), output))
det_names = pd.Series(imlist).apply(lambda x: "{}/det_{}".format(args.det,x.split("/")[-1]))
list(map(cv2.imwrite, det_names, orig_ims))
Код фактически создает новую папку с именем «det», как и при размещении там новых изображений, но после запуска модели папка все еще пуста.
Я попытался выполнить несколько операторов print до и после последней строки кода, который я здесь написал, и кажется, что код работает без сбоев, но он по-прежнему не сохраняет новые изображения.