最新推送

ultralytics_yolo11-main/train_resize_cls_muju.py

@@ -0,0 +1,19 @@
+from ultralytics import YOLO
+
+if __name__ == '__main__':
+    model = YOLO(r'/home/hx/yolo/muju_cls/yolo11-cls-muju-resize.yaml')
+    #model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/runs/train/cls_resize/exp_cls6/weights/best.pt')
+    results = model.train(
+        data='/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/模具车分类resize',
+        epochs=500,
+        imgsz=640,
+        batch=4,
+        workers=10,
+        device='0',
+        project='runs/train/cls_resize_muju',
+        name='exp_cls',
+        exist_ok=False,
+        optimizer='AdamW',
+        lr0=0.0005,
+        patience=0,
+    )