songbingle/le-yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[MF]修改文件

Browse Source
This commit is contained in:
songbingle 2025-06-06 18:05:09 +08:00
parent 3c5e3abfa9
commit f2bd169955
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/data.yaml
View File

@ -1,4 +1,4 @@
path: C:\workspace\le-yolo\data
path: /home/le/le-yolo/data
train: images/train
val: images/val
test: images/test

2
src/train.py
View File

@ -3,7 +3,7 @@ import time
start_time = time.time()
model = YOLO("yolov8n.pt")
# model = YOLO(r"C:\workspace\le-yolo\runs\detect\train40\weights\last.pt")
model.train(data="data.yaml", epochs=100, batch=8, device='cpu', imgsz=640, augment = True)
model.train(data="data.yaml", epochs=100, batch=8, device=0, imgsz=640, augment = True)
# model.train(data="data.yaml", epochs=100, batch=8, device='cpu', imgsz=640, augment = True, lr = 0.001,wight_decay = 0.0005 )
model.val()
print('训练完成')