增加部署代码
This commit is contained in:
@ -4,7 +4,7 @@ if __name__ == '__main__':
|
||||
model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/ultralytics/cfg/models/11/yolo11-obb.yaml')
|
||||
results = model.train(
|
||||
data='obb_data.yaml',
|
||||
epochs=100,
|
||||
epochs=1000,
|
||||
imgsz=640,
|
||||
batch=4,
|
||||
workers=10,
|
||||
@ -14,5 +14,4 @@ if __name__ == '__main__':
|
||||
exist_ok=False,
|
||||
optimizer='AdamW',
|
||||
lr0=0.001,
|
||||
patience=20,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user