Yolov3_resnet18的backbone使用resnet18,在调试过程中为了方便进行backbone的调试,附带了单独使用resnet18进行分类训练的能力。 后续修改中直接使用num_class作为分类训练/检测训练的区分,在二次开发中容易造成误解。 解决方案 添加feature_only参数,用来控制是否只导出feature_map用于完成后续的检测任务。 !21736:Add arg...
detectron2 backbone: resnet18, efficientnet, hrnet, mobilenet v2, resnest, bifpn - sxhxliang/detectron2_backbone