迁移学习方法
(1)预训练模型与训练模型的构建
- 实现思路
使用yolov8x.yaml的模型配置作为教师模型,学生模型采用yolov8n.yaml。先使用较大的教师模型在数据集上面进行训练,之后再用学生模型的网络将教师模型训练成为学生模型。
- 预训练模型参数
模型结构:268 layers
参数数量:68125494 parameters
计算性能:257.4 GFLOPs
- 训练模型参数
模型结构:400 layers
参数数量:1336652 parameters
计算性能:13.2 GFLOPs