search

IDEA-Research / detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.
https://detrex.readthedocs.io/en/latest/
Apache License 2.0
1.97k stars 206 forks source link

Provide more checkpoints. #207

Closed YiqunChen1999 closed 1 year ago

YiqunChen1999 commented 1 year ago

Hi, thanks for your great project.

Do you have plans to convert checkpoints (e.g., DAB-DETR-R50-DC5) from other repos of IDEA-Research?

Thanks.

rentainhe commented 1 year ago

Hi, thanks for your great project.

Do you have plans to convert checkpoints (e.g., DAB-DETR-R50-DC5) from other repos of IDEA-Research?

Thanks.

Sure, if you need this checkpoint, we would like to convert it and add it into detrex model zoo @YiqunChen1999

rentainhe commented 1 year ago

Hi, thanks for your great project.

Do you have plans to convert checkpoints (e.g., DAB-DETR-R50-DC5) from other repos of IDEA-Research?

Thanks.

We've provided the checkpoint converter to convert the weights from original repo, you can also check this documentation for mode details: https://detrex.readthedocs.io/en/latest/tutorials/Converters.html

YiqunChen1999 commented 1 year ago

Hi, thanks for your great project. Do you have plans to convert checkpoints (e.g., DAB-DETR-R50-DC5) from other repos of IDEA-Research? Thanks.

Sure, if you need this checkpoint, we would like to convert it and add it into detrex model zoo @YiqunChen1999

Sure, I have tried to convert DAB-DETR-R50-DC5, but the converted model only gives 4.35 AP. It would be helpful if you provide this checkpoint.

BTW, it would save much time for the community if it's possible to provide other checkpoints.

Thank you!

rentainhe commented 1 year ago

Hi, thanks for your great project. Do you have plans to convert checkpoints (e.g., DAB-DETR-R50-DC5) from other repos of IDEA-Research? Thanks.

Sure, if you need this checkpoint, we would like to convert it and add it into detrex model zoo @YiqunChen1999

Sure, I have tried to convert DAB-DETR-R50-DC5, but the converted model only gives 4.35 AP. It would be helpful if you provide this checkpoint.

BTW, it would save much time for the community if it's possible to provide other checkpoints.

Thank you!

OK, we will try to convert it for the community

alrightkami commented 1 year ago

It would be helpful as well to have a MaskDINO with Swin backbones.

rentainhe commented 1 year ago

It would be helpful as well to have a MaskDINO with Swin backbones.

Sure, we will refine MaskDINO project recently, and the checkpoint convert will be finished in this PR: https://github.com/IDEA-Research/detrex/pull/208 @alrightkami @YiqunChen1999

We also open an issue to collect the converters we used for converting weights: https://github.com/IDEA-Research/detrex/issues/209

rentainhe commented 1 year ago

Hi, thanks for your great project. Do you have plans to convert checkpoints (e.g., DAB-DETR-R50-DC5) from other repos of IDEA-Research? Thanks.

Sure, if you need this checkpoint, we would like to convert it and add it into detrex model zoo @YiqunChen1999

Sure, I have tried to convert DAB-DETR-R50-DC5, but the converted model only gives 4.35 AP. It would be helpful if you provide this checkpoint.

BTW, it would save much time for the community if it's possible to provide other checkpoints.

Thank you!

The temperature in position embedding should set to 10 in DAB-DETR-R50-DC5 to get the right testing AP~

rentainhe commented 1 year ago

I'm closing this issue~ feel free to reopen it if needed