Update python_scripts/DLM_Tree_Classification/DLM_training_script

dda6b339 · Mayer · d0af6ca2 · dda6b339
Commit dda6b339 authored 7 months ago by Mayer
Hide whitespace changes
Inline Side-by-side

Showing

with 1 addition and 1 deletion
+1 -1
--- a/python_scripts/DLM_Tree_Classification/DLM_training_script
+++ b/python_scripts/DLM_Tree_Classification/DLM_training_script
@@ -47,7 +47,7 @@ names: ['Großer Laubbaum',
 model = YOLO("yolov8n.pt")

 # Train the model
-results = model.train(data="sorted_images_YOLO_formatted/data/yolov8.yaml", epochs=500, imgsz=200, cache=False)  
+results = model.train(data="sorted_images_YOLO_formatted/data/yolov8.yaml", epochs=300, imgsz=200, cache=False)