'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