将yolov5识别出的文件夹中的图片截取并保存到指定文件夹

以下是 Python 代码示例，可以将 yolov5 识别出的文件夹中的图片截取并保存到指定的文件夹：

import os
import cv2
 
# 设置输入输出文件夹路径
input_folder_path = 'path/to/yolov5/output/folder'
output_folder_path = 'path/to/save/cropped/images'
 
# 创建输出文件夹
if not os.path.exists(output_folder_path):
    os.makedirs(output_folder_path)
 
# 加载文件夹中的所有文件名，以 .jpg 结尾的文件将被处理
file_names = os.listdir(input_folder_path)
file_names = [f for f in file_names if f.endswith('.jpg')]
 
# 循环处理每个文件
for file_name in file_names:
    # 读取图像
    image_path = os.path.join(input_folder_path, file_name)
    image = cv2.imread(image_path)
 
    # 定义感兴趣区域，截取图像，并保存到输出文件夹
    x, y, w, h = （#定义感兴趣区域）
    cropped_image = image[y:y+h, x:x+w]
    output_path = os.path.join(output_folder_path, file_name)
    cv2.imwrite(output_path, cropped_image)

要运行这个程序，需要安装 OpenCV 库：

pip install opencv-python

请注意，需要在代码中定义感兴趣区域。如果你使用 yolov5 训练一个对象检测器，那么它将识别出边界框，你可以使用这些边界框来定义感兴趣区域，并对它们进行裁剪。如果没有边界框，则需要根据图像中对象的位置定义感兴趣区域。