yolov5

 格式转png

from PIL import Image
import os

src_folder  = "dataset"
target_folder = "new_data"

for filen_name in os.listdir(src_folder):
    if filen_name.endswith(".jpg") or filen_name.endswith(".JPG"):
        im = Image.open(os.path.join(src_folder,filen_name))
        new_filename = os.path.splitext(filen_name)[0]+".png"
        print(new_filename)
        im.save(os.path.join(target_folder,new_filename),"png")

结果转onnx

可以labelme2onnox就不需要

import torch
from models.experimental import attempt_load

#加载模型权重
model = attempt_load("D:/Projects/00_Image/Image Detection/YOLOv5/yolov5-master/runs/train/exp4/weights/best.pt",device=torch.device("cpu"))
#设置为评估模式
model.eval()
#准备一个示例输入
input_tensor = torch.randn(1,3,640,640)
#导出模型
torch.onnx.export(model,input_tensor,"yolov5s.onnx")