search

facebookresearch / adaptive_teacher

This repo provides the source code for "Cross-Domain Adaptive Teacher for Object Detection".
Other
180 stars 35 forks source link

backbone modification #58

Open achintski opened 1 year ago

achintski commented 1 year ago

Dear sir, Thank you for sharing your work！ I have a question about how to modify the backbone model. The backbone used in your code is R101 or VGG16, and I want to modify it to Yolov7's backbone. Since I am an undergraduate and just getting started with object detection, I am not familiar with modifying the code. I kindly ask you to provide some ideas for modification (backbone part and train part), and I will modify it by myself according to your ideas. Looking forward to hearing from you！ Thanks in advance！

yujheli commented 1 year ago

Register the backbone following my version of VGG: https://github.com/facebookresearch/adaptive_teacher/blob/main/adapteacher/modeling/meta_arch/vgg.py The Detectron2 usually supports only ResNet so you will need to register your own. It is not easy to follow the code in Detectron2 but would be necessary to trace the code.