Commit
•
966a6ef
1
Parent(s):
d28974a
retrain model
Browse files
taekwondo_motion_learning/machine_learning/image_classification/runs/classify/train/weights/best.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 2964747
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e48624c2ccb902fb620918cb39c5f4a5f20a3ba16ce9a75f8e7f525f594275fc
|
| 3 |
size 2964747
|
taekwondo_motion_learning/machine_learning/image_classification/runs/classify/train/weights/last.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 2966859
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c81eb1e09a5301d5112042949edc990a5df4ad89c901f7d980aa0ac959932a9b
|
| 3 |
size 2966859
|
taekwondo_motion_learning/machine_learning/image_classification/train.py
CHANGED
|
@@ -1,10 +1,8 @@
|
|
| 1 |
-
from ultralytics import YOLO
|
| 2 |
import shutil
|
| 3 |
|
| 4 |
shutil.rmtree('./runs', ignore_errors=True)
|
| 5 |
|
| 6 |
-
settings.update({'runs_dir': './runs'})
|
| 7 |
-
|
| 8 |
model = YOLO('yolov8n-cls.pt') # Pretrained model (recommended for training)
|
| 9 |
|
| 10 |
model.train(data='./dataset', epochs=50, imgsz=64)
|
|
|
|
| 1 |
+
from ultralytics import YOLO
|
| 2 |
import shutil
|
| 3 |
|
| 4 |
shutil.rmtree('./runs', ignore_errors=True)
|
| 5 |
|
|
|
|
|
|
|
| 6 |
model = YOLO('yolov8n-cls.pt') # Pretrained model (recommended for training)
|
| 7 |
|
| 8 |
model.train(data='./dataset', epochs=50, imgsz=64)
|