yolov7_obb+TensorRT+WIN10

github链接

GitHub - lx-cly/YOLOv7_OBB: YOLOv7+KLD

训练命令:

python train.py --weights runs/yolov7obb/base4/weights/last.pt --cfg cfg/deploy/yolov7.yaml --img-size 1024 --batch-size 10 --epochs 1500 --project runs/yolov7obb/ --device 0,1,2,3 --resume runs/yolov7obb/base4/weights/last.pt

数据集格式--具体格式和之前训练的yolov5_obb一致。

Win10推理+导出:

猜你喜欢

转载自blog.csdn.net/zhangdaoliang1/article/details/129259645