diff --git a/algorithm/data.yaml b/algorithm/data.yaml new file mode 100644 index 0000000..8d18830 --- /dev/null +++ b/algorithm/data.yaml @@ -0,0 +1,6 @@ +path: dataset +train: images\train +val: images\val +device: 0 +names: + 0: 人 diff --git a/algorithm/train.py b/algorithm/train.py new file mode 100644 index 0000000..9b038ec --- /dev/null +++ b/algorithm/train.py @@ -0,0 +1,11 @@ +from ultralytics import YOLO + +model = YOLO("yolo11n.pt") + +model.train( + data="data.yaml", + epochs=100, + imgsz=640, + batch=16, + name="train1" +) \ No newline at end of file diff --git a/algorithm/yolo11n.pt b/algorithm/yolo11n.pt new file mode 100644 index 0000000..45b273b Binary files /dev/null and b/algorithm/yolo11n.pt differ