如何将检测到的对象Yolo-v3保存为新图像?

时间:2019-06-01 17:21:30

标签: opencv machine-learning pytorch object-detection yolo

我正在遵循此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”的新文件夹,就像它应该将新图像放置在其中一样,但是在模型运行之后,该文件夹仍然为空。
我试图在我在此处编写的代码的最后一行之前和之后运行一些打印语句,似乎代码运行平稳,但仍然无法保存新图像。

0 个答案:

没有答案