lichen/ieemoo-ai-gift
1
0
Fork 0
mirror of https://gitee.com/nanjing-yimao-information/ieemoo-ai-gift.git synced 2025-08-23 23:50:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

增加测试维度

Browse Source
This commit is contained in:
lee
2025-06-25 16:23:20 +08:00
parent 3d364aa0bd
commit 96a9445761
6 changed files with 141 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -17,5 +17,5 @@ model = YOLOv10('ckpts/weights/yolov10n.pt')
#model.train(data='coco.yaml', epochs=1, batch=64, imgsz=640)
#model.train(data='coco128_cls10_0924.yaml', epochs=300, batch=64, imgsz=640, resume=False)
#model.train(data='coco128_cls10_1010.yaml', epochs=300, batch=128, imgsz=640, resume=False)
model.train(data='gift.yaml', epochs=400, batch=32, imgsz=224, resume=False, save_dir='/ckpts')
model.train(data='gift.yaml', epochs=600, batch=32, imgsz=224, resume=False, save_dir='/ckpts')
#model.train(data='coco128_cls10_1010_1205.yaml', epochs=300, batch=32, imgsz=640, resume=True)