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

[Feature] Add feature refine CUDA op for rotated detection. #1603

Merged
merged 16 commits into from
Dec 23, 2021

Conversation

zytx121
Copy link
Contributor

@zytx121 zytx121 commented Dec 21, 2021

Motivation

Add feature refine CUDA operation for rotate detection.

@inproceedings{yang2021r3det,
    title={R3Det: Refined Single-Stage Detector with Feature Refinement for Rotating Object},
    author={Yang, Xue and Yan, Junchi and Feng, Ziming and He, Tao},
    booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
    volume={35},
    number={4},
    pages={3163--3171},
    year={2021}
}

Modification

Several files in mmcv/ops folder.

BC-breaking (Optional)

No.

Use cases (Optional)

from mmcv.ops import feature_refine

zytx121 and others added 5 commits December 21, 2021 17:00
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
@zhouzaida zhouzaida requested a review from ZwwWayne December 22, 2021 03:01
@zytx121 zytx121 changed the title [Feature] Add feature refine ops for rotate detection. [Feature] Add feature refine ops for rotated detection. Dec 22, 2021
@zytx121 zytx121 changed the title [Feature] Add feature refine ops for rotated detection. [Feature] Add feature refine op for rotated detection. Dec 22, 2021
@zytx121 zytx121 changed the title [Feature] Add feature refine op for rotated detection. [Feature] Add feature refine CUDA op for rotated detection. Dec 22, 2021
zytx121 and others added 8 commits December 22, 2021 15:10
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
@zhouzaida
Copy link
Collaborator

Unit test failed in my local machine.
image
More details can be found at pytorch/pytorch#16940.

@zytx121
Copy link
Contributor Author

zytx121 commented Dec 23, 2021

Unit test failed in my local machine. image More details can be found at pytorch/pytorch#16940.

Thanks for your web link. We have fixed it.

@ZwwWayne ZwwWayne merged commit 2475dc3 into open-mmlab:master Dec 23, 2021
@zhouzaida zhouzaida added the CUDA label Dec 25, 2021
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

Successfully merging this pull request may close these issues.

3 participants