wudashuo / yolov5

YOLOv5 汉化版,保持官方同步更新
GNU General Public License v3.0
445 stars 100 forks source link

请问您知道如何yolov5在训练中如何加入标签平滑(label smoothing)技巧吗? #19

Open Wanghe1997 opened 2 years ago

Wanghe1997 commented 2 years ago

Search before asking

Question

在代码中如何设置并实现呢?谢谢

Additional

No response

github-actions[bot] commented 2 years ago

👋 你好 @Wanghe1997, 如有任何问题,请首先检查你的运行指令有没有问题,如果指令没有问题,请尝试更新作者仓库的最新代码:

  # 如果没下载官方代码
  $ git clone https://github.com/ultralytics/yolov5.git
  $ cd yolov5
  $ pip install -r requirements.txt
  # 如果已下载官方代码
  $ cd yolov5
  $ git reset --hard
  $ git pull
  $ pip install -r requirements.txt

更多请参考⭐️英文官方教程

依赖

Python版本3.6或更高,python依赖库都在requirements.txt 里面,直接pip install -r requirements.txt即可。 如果你使用Windows的话,尽量使用CUDA10.2和对应版本的pytorch,CUDA11+会有些许问题。

环境

下面是已经配置好环境的免费GPU训练环境:

wudashuo commented 2 years ago

代码是现成的,只是默认是0没开启而已

python train.py --label-smoothing 0.01

参数根据你的数据集自己慢慢调吧