train.py 274 B

1234567
  1. from ultralytics import YOLO
  2. # Load an official or custom model
  3. #model = YOLO('yolov8s-seg.pt') # Load an official Detect model
  4. model = YOLO('fivefingers-seg-100.pt') # Load an official Detect model
  5. results = model.train(data='data.yaml', epochs=100, imgsz=640, batch=8)