Преглед на файлове

上傳檔案到 'segment'

fatwolf преди 8 месеца
родител
ревизия
9cfa31a406
променени са 1 файла, в които са добавени 8 реда и са изтрити 0 реда
  1. 8 0
      segment/seg_train.py

+ 8 - 0
segment/seg_train.py

@@ -0,0 +1,8 @@
+from ultralytics import YOLO
+
+# Load a model
+model = YOLO('yolov8n-seg.pt')  # build a new model from YAML
+
+
+# Train the model
+results = model.train(data='seg_magnet.yaml', epochs=150, imgsz=640)