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="" #必须设置,否则还会有旋转、平移
                         )

 

posted @ 2024-12-26 16:03  夕西行  阅读(611)  评论(0)    收藏  举报