Jinhang Zhu comments

Repositories
Issues
Comments

Results 2 comments of


                                            Jinhang Zhu

gpu softnms is slower because you use numpy in the middle of a torch ops

@zylo117 Hi, I implemented the pure pytorch version of the softnms function and I currently write it in the google colab file: https://colab.research.google.com/drive/1gzhXX-LyMdZ41qHv0rKzHpxKLYliEiPn?usp=sharing What made me confused is that GPU...

gpu softnms is slower because you use numpy in the middle of a torch ops

@zylo117 Thank you for your explanation. I think I will try to improve the algorithm in the future.