search

tztztztztz / eql.detectron2

The official implementation of Equalization Loss for Long-Tailed Object Recognition (CVPR 2020) based on Detectron2. https://arxiv.org/abs/2003.05176
Apache License 2.0
202 stars 17 forks source link

How to get caffe model and apply into your code? #11

Closed Chauncy-Cai closed 3 years ago

Chauncy-Cai commented 3 years ago

I would like to get the result you report. But I CANNOT with the pretrained model provided by detectron2. Can you provide the link to the cafe model(ResNet-50) you say in README? Moreover, I am not sure whether detectron2 has code to convert caffe-model to .pth, or detectron2 can use caffe-model directly. Can you show me the code? Or the convert code written in detectron1 is OK?

tztztztztz commented 3 years ago

Here is the link for caffe model: https://github.com/facebookresearch/detectron2/blob/master/MODEL_ZOO.md#imagenet-pretrained-models

Detectron2 can load the caffe model directly.