Jisnu Kalita Claude Sonnet 4.6 commited on
Commit
813519c
·
1 Parent(s): bb3c9cb

fine-tune from best.pt, 75 epochs, drop drone images

Browse files

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -8,11 +8,11 @@ RUNS_DIR = Path(__file__).parent / "runs"
8
 
9
 
10
  def train():
11
- model = YOLO("yolo11x.pt")
12
 
13
  results = model.train(
14
  data=str(DATASET_YAML),
15
- epochs=200,
16
  imgsz=1280,
17
  batch=128,
18
  project=str(RUNS_DIR),
 
8
 
9
 
10
  def train():
11
+ model = YOLO("runs/pothole/weights/best.pt")
12
 
13
  results = model.train(
14
  data=str(DATASET_YAML),
15
+ epochs=75,
16
  imgsz=1280,
17
  batch=128,
18
  project=str(RUNS_DIR),