将YOLOv5训练得到的pth文件转换成pt文件
时间: 2023-03-28 18:00:45 浏览: 1013
yolov5官方训练模型PT
5星 · 资源好评率100%
可以使用以下命令将YOLOv5训练得到的pth文件转换成pt文件:
python models/export.py --weights /path/to/yolov5s.pt --img 640 --batch 1
其中,/path/to/yolov5s.pt是训练得到的pth文件的路径,--img和--batch参数可以根据需要进行调整。
阅读全文