怎么修改yolov5中的detect源码,将检测出来的图片和他的原图分别保存在不同的文件夹
时间: 2024-02-27 22:53:17 浏览: 20
要修改Yolov5中的detect源码,将检测出来的图片和它的原图分别保存在不同的文件夹中,可以按照以下步骤进行:
1. 打开detect.py文件,这个文件在Yolov5的代码库中的models目录下。
2. 找到detect()函数,这个函数是用来进行物体检测的核心代码。
3. 在函数的代码中找到将检测出的图片保存到本地的代码,通常是类似于下面这样的代码:
```
cv2.imwrite(save_path, im0)
```
4. 在这行代码之前,添加如下代码,来保存原图和检测出的图片到不同的文件夹中:
```
import os
# 创建保存原图的文件夹
if not os.path.exists('orig_images'):
os.makedirs('orig_images')
# 创建保存检测出的图片的文件夹
if not os.path.exists('detected_images'):
os.makedirs('detected_images')
# 保存原图
cv2.imwrite(os.path.join('orig_images', os.path.basename(image_path)), img)
# 保存检测出的图片
cv2.imwrite(os.path.join('detected_images', os.path.basename(save_path)), im0)
```
这样,就可以将原图和检测出的图片分别保存到不同的文件夹中了。注意,需要先创建好保存文件的文件夹,可以根据实际情况进行修改。
5. 保存修改后的代码,并重新运行程序,就可以看到保存了原图和检测出的图片的结果了。
相关推荐
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)