'add_algorithm'
This commit is contained in:
parent
7489473ba4
commit
12a8f7426c
6
algorithm/data.yaml
Normal file
6
algorithm/data.yaml
Normal file
@ -0,0 +1,6 @@
|
|||||||
|
path: dataset
|
||||||
|
train: images\train
|
||||||
|
val: images\val
|
||||||
|
device: 0
|
||||||
|
names:
|
||||||
|
0: 人
|
11
algorithm/train.py
Normal file
11
algorithm/train.py
Normal file
@ -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"
|
||||||
|
)
|
BIN
algorithm/yolo11n.pt
Normal file
BIN
algorithm/yolo11n.pt
Normal file
Binary file not shown.
Loading…
x
Reference in New Issue
Block a user