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 filesCo-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
main.py
CHANGED
|
@@ -8,11 +8,11 @@ RUNS_DIR = Path(__file__).parent / "runs"
|
|
| 8 |
|
| 9 |
|
| 10 |
def train():
|
| 11 |
-
model = YOLO("
|
| 12 |
|
| 13 |
results = model.train(
|
| 14 |
data=str(DATASET_YAML),
|
| 15 |
-
epochs=
|
| 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),
|