增加cvat反向上传
This commit is contained in:
19
ultralytics_yolo11-main/train_xiantiao_detect.py
Normal file
19
ultralytics_yolo11-main/train_xiantiao_detect.py
Normal file
@ -0,0 +1,19 @@
|
||||
from ultralytics import YOLO
|
||||
|
||||
if __name__ == '__main__':
|
||||
#model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/ultralytics/cfg/models/11/yolo11-obb.yaml')
|
||||
model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/ultralytics/cfg/models/11/yolo11.yaml')
|
||||
results = model.train(
|
||||
data='xiantiaodata.yaml',
|
||||
epochs=2000,
|
||||
imgsz=640,
|
||||
batch=4,
|
||||
workers=10,
|
||||
device='0',
|
||||
project='runs/train',
|
||||
name='exp_detect',
|
||||
exist_ok=False,
|
||||
optimizer='AdamW',
|
||||
lr0=0.0005,
|
||||
patience=0,
|
||||
)
|
||||
Reference in New Issue
Block a user