fix format, multiple objects
This commit is contained in:
parent
8951dd27f0
commit
642ac4033c
3
.gitignore
vendored
3
.gitignore
vendored
@ -4,6 +4,9 @@
|
||||
# Datensätze (ganzer Ordner)
|
||||
datasets/
|
||||
|
||||
# Datensätze (ganzer Ordner)
|
||||
runs/
|
||||
|
||||
# Große Model-Dateien
|
||||
*.pt
|
||||
|
||||
|
||||
6
main.py
6
main.py
@ -6,13 +6,13 @@ from ultralytics import YOLO
|
||||
#model = YOLO("yolo11n.yaml")
|
||||
|
||||
# Load a pretrained YOLO model (recommended for training)
|
||||
#model = YOLO("yolo11n-pose.pt")
|
||||
model = YOLO("yolo11n-pose.pt")
|
||||
|
||||
# Train the model using the 'coco8.yaml' dataset for 3 epochs
|
||||
#results = model.train(data="datasets/skier_pose/skier_pose.yaml", epochs=200, imgsz=640, device='mps')
|
||||
results = model.train(data="datasets/skier_pose/skier_pose.yaml", epochs=200, imgsz=640, device='mps')
|
||||
|
||||
# Evaluate the model's performance on the validation set
|
||||
#results = model.val()
|
||||
results = model.val()
|
||||
|
||||
# Perform object detection on an image using the model
|
||||
#results = model("bus.jpg", device='cpu')
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue
Block a user