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)
|
# Datensätze (ganzer Ordner)
|
||||||
datasets/
|
datasets/
|
||||||
|
|
||||||
|
# Datensätze (ganzer Ordner)
|
||||||
|
runs/
|
||||||
|
|
||||||
# Große Model-Dateien
|
# Große Model-Dateien
|
||||||
*.pt
|
*.pt
|
||||||
|
|
||||||
|
|||||||
6
main.py
6
main.py
@ -6,13 +6,13 @@ from ultralytics import YOLO
|
|||||||
#model = YOLO("yolo11n.yaml")
|
#model = YOLO("yolo11n.yaml")
|
||||||
|
|
||||||
# Load a pretrained YOLO model (recommended for training)
|
# 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
|
# 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
|
# Evaluate the model's performance on the validation set
|
||||||
#results = model.val()
|
results = model.val()
|
||||||
|
|
||||||
# Perform object detection on an image using the model
|
# Perform object detection on an image using the model
|
||||||
#results = model("bus.jpg", device='cpu')
|
#results = model("bus.jpg", device='cpu')
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue
Block a user