; 1. 知识蒸馏训练
1) 模型准备
学生模型 yolov5s和教师模型yolov5m 模型准备
知识蒸馏训练，需要用到 yolov5s和 yolov5m baseline的训练结果，把 runs/train/yolov5s-baseline/weights/best.pt和 runs/train/yolov5m-baseline/weights/best.pt拷贝到 weights文件夹下并改名为 yolov5s_bm.pt和 yolov5m_bm.pt
注意batch-size大时可能有显存溢出
; 2) 知识蒸馏训练
训练命令
python train_distillation.py --weights weights/yolov5s_bm.pt --cfg
models
Original: https://blog.csdn.net/weixin_38346042/article/details/126072439
Author: @BangBang
Title: 知识蒸馏6: yolov5知识蒸馏训练
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/648892/
转载文章受原作者版权保护。转载请注明原作者出处！