Skip to content
This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

the Retinanet is not available right now? #513

Open
feihuidiqiu opened this issue Mar 1, 2019 · 6 comments
Open

the Retinanet is not available right now? #513

feihuidiqiu opened this issue Mar 1, 2019 · 6 comments
Labels

Comments

@feihuidiqiu
Copy link

❓ Questions and Help

I test the Retinanet by replace the
config_file = "../configs/caffe2/e2e_mask_rcnn_R_50_FPN_1x_caffe2.yaml"
to
config_file = "../configs/retinanet/retinanet_R-50-FPN_1x.yaml"
in Mask_R-CNN_demo.ipynb,
and then i get 0 bbox in the result, is the Retinanet under develop or i need to change the input of picture, thank you for any help~

@fmassa
Copy link
Contributor

fmassa commented Mar 1, 2019

Hi,

We haven't uploaded pre-trained models for RetinaNet.

@chengyangfu do you know if the pre-trained models from Caffe2 work with the version in here, or do we need to modify the c2_model_loading for that?

@fmassa fmassa added enhancement New feature or request contributions welcome labels Mar 1, 2019
@feihuidiqiu
Copy link
Author

feihuidiqiu commented Mar 1, 2019

Hi,

We haven't uploaded pre-trained models for RetinaNet.

@chengyangfu do you know if the pre-trained models from Caffe2 work with the version in here, or do we need to modify the c2_model_loading for that?

Thank you,
I didn't notice that.

@chengyangfu
Copy link
Contributor

@fmassa
Unfortunately, we need to modify the model from Caffe2 when loading it. I will check that part this weekend.

@feihuidiqiu,
You can find some RetinaNet models trained on this repository here. #102
You just need to change cfg.merge_from_list(["MODEL.DEVICE", "cpu"]) to
cfg.merge_from_list(["MODEL.DEVICE", "cpu", "MODEL.WEIGHT", "THE_MODEL_YOU_DOWNLOAD"])

@fmassa
Copy link
Contributor

fmassa commented Mar 1, 2019

@Computational-Camera
Copy link

Actually, after 36 hours' training with a single GPU, the results from Retinanet does not make me disappointed.

@zxhr2793
Copy link

Actually, after 36 hours' training with a single GPU, the results from Retinanet does not make me disappointed.

Which backbone and what are the results?

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
Projects
None yet
Development

No branches or pull requests

5 participants