知识蒸馏6: yolov5知识蒸馏训练

1. 知识蒸馏训练

1) 模型准备

学生模型 yolov5s和教师模型yolov5m 模型准备

知识蒸馏训练,需要用到yolov5syolov5m baseline的训练结果,把runs/train/yolov5s-baseline/weights/best.ptruns/train/yolov5m-baseline/weights/best.pt拷贝到weights文件夹下并改名为yolov5s_bm.ptyolov5m_bm.pt
知识蒸馏6: yolov5知识蒸馏训练_第1张图片
注意batch-size大时可能有显存溢出

2) 知识蒸馏训练

训练命令

python train_distillation.py --weights weights/yolov5s_bm.pt --cfg 
models

你可能感兴趣的:(模型轻量化,深度学习,人工智能,神经网络)