tuklu
/

pothole_detection

Model card Files Files and versions

Jisnu Kalita commited on 17 days ago

Commit

6124b76

·

1 Parent(s): 2fd47db

model update

Files changed (1) hide show

main.py +2 -2

main.py CHANGED Viewed

@@ -8,11 +8,11 @@ RUNS_DIR = Path(__file__).parent / "runs"
 def train():
-    model = YOLO("yolo11n.pt")
     results = model.train(
         data=str(DATASET_YAML),
-        epochs=50,
         imgsz=640,
         batch=16,
         project=str(RUNS_DIR),

 def train():
+    model = YOLO("yolo11x.pt")
     results = model.train(
         data=str(DATASET_YAML),
+        epochs=200,
         imgsz=640,
         batch=16,
         project=str(RUNS_DIR),