Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Is there an implementation based on the original DETR code? #4

Open
linhuixiao opened this issue Aug 4, 2023 · 1 comment
Open

Comments

@linhuixiao
Copy link

linhuixiao commented Aug 4, 2023

Thanks for your nice work, I am going to try to use your proposed method in our work, but the current code repository code is based on the implementation of detectron v2, the implementation of the function call is a bit complicated, is there an implementation based on the original DETR code? Or can you explain the key code of refinebox? I think an implementation based on the original DETR might help to increase the impact of this work.

@YiqunChen1999
Copy link
Owner

Thanks for your attention. We don't have an implementation based on the original DETR code.

RefinerMixin describes the interface. The forward_refinement will be called after the detector produces detection results. This method refines the detector outputs (by calling the refine_box method) and returns loss_dict (training) or refined results (inference).

As for the refinement network, please see res_refiner.py and fpn.py

I hope this will be helpful.

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

No branches or pull requests

2 participants