摘要: 1、魔改的yolo11-cls(主改head.py内容),加载时使用如下方式,否则可能魔改无效 yolo = YOLO("yolo11n-cls.yaml").load("yolo11n-cls.pt") yolo11n-cls.yaml里将nc写成自己的类别数 2、train时data指定的yam 阅读全文
posted @ 2024-12-15 20:23 夕西行 阅读(17) 评论(0) 推荐(0) 编辑
摘要: 当使用export,dynamic=True时,batch和尺寸都是动态的。 若想只有batch动态,修改下图指示代码即可 yolo.export(format="onnx",imgsz=(224,224),dynamic=True) 阅读全文
posted @ 2024-12-15 12:55 夕西行 阅读(10) 评论(0) 推荐(0) 编辑