Skip to content

optimize nms#1

Open
chunyuan-w wants to merge 2 commits intoXiaobingSuper:int8from
chunyuan-w:optimize_nms
Open

optimize nms#1
chunyuan-w wants to merge 2 commits intoXiaobingSuper:int8from
chunyuan-w:optimize_nms

Conversation

@chunyuan-w
Copy link

Port nms from python to c++ and use parallel for to optimize it.

@chunyuan-w
Copy link
Author

@XiaobingSuper Please help to review it.

Do I need to remove as well the .to("cpu") in the decoding part of the calibration?
https://github.com/XiaobingSuper/inference/blob/9c6c40367fd4652ce35d5eb87357611cc5c10f2f/others/cloud/single_stage_detector/pytorch/infer.py#L148

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.

1 participant