step 1.环境准备
step 2.训练一个学生网络(以训练coco数据集为例)
step3.训练一个教师网络
step 4 准备知识蒸馏训练
4.1 修改train_distillation.py
4.2 运行train_distillation.py
step 5 成功训练
step 6.蒸馏代码
step 1.环境准备
运行下面语句准备yolov5运行环境
pip install -r requirements.txt
注:若环境已经安装好可不执行,直接将step 4中的 train_distillation.py,放置在与yolov5中的train.py文件的同一级目录下即可,we

step 2.训练一个学生网络(以训练coco数据集为例)
python train.py --data data/coco128.yaml --cfg models/yolov5s.yaml --weights yolov5s.pt--batch-size 8 --epochs 30