search

irfanICMLL / structure_knowledge_distillation

The official code for the paper 'Structured Knowledge Distillation for Semantic Segmentation'. (CVPR 2019 ORAL) and extension to other tasks.
BSD 2-Clause "Simplified" License
695 stars 104 forks source link

Training object detection model #5

Closed ivbelkin closed 4 years ago

ivbelkin commented 4 years ago

Great work!

I see, you present results on COCO object detection, but I cant find training procedure and which loss function did you use. Could you share it please?

Thank you!

irfanICMLL commented 4 years ago

I just use the code of FCOS: https://github.com/tianzhi0549/FCOS , and you can modify their code by adding the distillation loss we provided. I use the output logits and the features from all the towers. If you have any problems, please feel free to contact me~