FGD针对目标检测的重点与全局知识蒸馏

CVPR 2022 | 针对目标检测的重点与全局知识蒸馏

文章链接:arxiv.org/abs/2111.11837
代码链接:github.com/yzd-v/FGD

讲解部分:https://zhuanlan.zhihu.com/p/477707304

代码部分

在网络计算损失部分

            #----------------------#
            #   前向传播
            #----------------------#
            outputs         = model_train(images)#学生网络的损失
            T_pred          = teacher(images)#教师网络的损失

            loss_value_all  

你可能感兴趣的:(知识蒸馏,Pytorch,目标检测,目标检测,深度学习,计算机视觉)