yolov8知识蒸馏代码详解:支持logit和feature-based蒸馏

文章目录

    • 1. 知识蒸馏理论
    • 2. yolov8 蒸馏代码应用
      • 2.1 环境配置
      • 2.2 训练模型
        • (1) 训练教师模型
        • (2) 训练学生模型baseline
        • (3) 蒸馏训练
    • 3. 知识蒸馏代码详解
      • 3.1 蒸馏参数设置
      • 3.2 蒸馏损失代码讲解
        • 3.2.1 Feature based loss
        • 3.2.1 Logit loss
      • 3.3 获取蒸馏的feature map及channels

你可能感兴趣的:(模型轻量化,yolov8,代码详解,知识蒸馏)