【知识蒸馏】开源：Yolov5模型知识蒸馏训练（可用于自己的数据集）_yolov5 蒸馏增加数据集的分类

作者：你好赵伟 | 2024-04-24 18:58:57

踩

yolov5 蒸馏增加数据集的分类

代码地址：

https://github.com/Sharpiless/Yolov5-distillation-train-inference

蒸馏训练：

python train_distill.py --weights yolov5s.pt \
    --teacher weights/yolov5l_voc.pt --distill_ratio 0.001 \
    -- teacher-cfg model/yolov5l.yaml --data data/voc.yaml \
    --epochs 30 --batch-size 16
1
2
3
4

训练参数:

–weights：预训练模型

–teacher：教师模型权重

–distill-ratio：蒸馏损失权重

实验结果：

数据集&#x

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/你好赵伟/article/detail/481062

【知识蒸馏】开源：Yolov5模型知识蒸馏训练（可用于自己的数据集）_yolov5 蒸馏 增加数据集的分类

代码地址：

蒸馏训练：

训练参数:

实验结果：

【知识蒸馏】开源：Yolov5模型知识蒸馏训练（可用于自己的数据集）_yolov5 蒸馏增加数据集的分类