部署文件+resize_cls+resize_seg
This commit is contained in:
18
ultralytics_yolo11-main/train_resize_cls.py
Normal file
18
ultralytics_yolo11-main/train_resize_cls.py
Normal file
@ -0,0 +1,18 @@
|
||||
from ultralytics import YOLO
|
||||
|
||||
if __name__ == '__main__':
|
||||
model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/ultralytics/cfg/models/11/yolo11-cls-resize.yaml')
|
||||
results = model.train(
|
||||
data='/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/classdata3',
|
||||
epochs=1000,
|
||||
imgsz=640,
|
||||
batch=4,
|
||||
workers=10,
|
||||
device='0',
|
||||
project='runs/train/cls_resize',
|
||||
name='exp_cls',
|
||||
exist_ok=False,
|
||||
optimizer='AdamW',
|
||||
lr0=0.001,
|
||||
patience=0,
|
||||
)
|
||||
Reference in New Issue
Block a user