19 lines
406 B
Python
19 lines
406 B
Python
|
from ultralytics import YOLO
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
model = YOLO(r'5cls.yaml')
|
||
|
|
results = model.train(
|
||
|
|
data='/home/dy/dataset/61_cls',
|
||
|
|
epochs=200,
|
||
|
|
imgsz=640,
|
||
|
|
batch=4,
|
||
|
|
workers=5,
|
||
|
|
device='0',
|
||
|
|
project='runs/train/61cls',
|
||
|
|
name='exp_cls',
|
||
|
|
exist_ok=False,
|
||
|
|
optimizer='AdamW',
|
||
|
|
lr0=0.001,
|
||
|
|
patience=0,
|
||
|
|
)
|