yolo11分类模型,禁用数据增强
由于平移、旋转、遮掩等可能将在边缘或很小的特征全部丢掉,因此禁用数据增强操作
yolo = YOLO("yolo11n-cls.yaml").load("yolo11n-cls.pt") results = yolo.train(data='/home/用户名/datasets/classify/dst', epochs=400, batch=32, imgsz=160, patience=0, translate=0.0, #不平移 scale=0.0,#不缩放 flipud=0.5, #上下翻转的概率 fliplr=0.5, #左右翻转的概率 erasing=0.0, #无遮挡 hsv_h=0.0, #不进行hsv操作 hsv_s=0.0, hsv_v=0.0, degrees=0.0, #不旋转 auto_augment="" #必须设置,否则还会有旋转、平移 )