在yolov5源码中哪里修改可以实现保存目标图片
时间: 2024-02-01 18:14:16 浏览: 53
yolov5目标检测代码yolov5-master (1).zip
在 YOLOv5 源码中,可以在 `detect.py` 文件中找到保存目标图片的代码。具体来说,可以在 `process_results()` 函数中找到以下代码段:
```python
if save_img: # Save results (optional)
save_path = str(Path(output) / Path(img_path).name)
plot_image(img, pred, save_path, classes, **kwargs)
```
其中,`plot_image()` 函数会将检测结果绘制在原始图片上,并保存到指定路径。可以通过修改 `save_path` 变量来指定保存路径,例如:
```python
save_path = str(Path(output) / Path("detected_" + img_path).name)
```
这样就会在 `output` 目录下保存以 "detected_" 为前缀的目标图片。
阅读全文