用yolov4自动分类图片，并将同类图片放入同一文件夹里

import cv2  
import numpy as np  
import os  
  
# Load YOLOv4 model  
net = cv2.dnn_DetectionModel('yolov4.cfg', 'yolov4.weights')  
net.setInputSize(416, 416)  # 设置输入图像的尺寸  
net.setInputScale(1.0 / 255)  # 对输入图像进行归一化  
net.setInputSwapRB(True)  # 交换图像的通道顺序  
  
# Load class labels  
classes = []  
with open('labels.txt', 'r') as f:  
    classes = [line.strip() for line in f.readlines()]  
  
# Load dataset  
dataset_path = "D:\projects\TestHello\pics"  #源图片位置
images = os.listdir(dataset_path)  
  
# Create output folder  
output_folder = "D:\projects\TestHello\output"  #图片输出问题
os.makedirs(output_folder, exist_ok=True)  
  
# Process each image  
for image_file in images:  
    image_path = os.path.join(dataset_path, image_file)  
    img = cv2.imread(image_path)  
  
    # Run inference  
    class_ids, confidences, boxes = net.detect(img, confThreshold=0.5, nmsThreshold=0.4)