Skip to content

Conversation

@ShinjoWang
Copy link

  • Change for-loop iou calculation to once
  • Change for-loop get-offset to once
  • Change number of classes in nms

In my case (Intel(R) Core(TM) i7-6850K CPU @ 3.60GHz, NVIDIA GeForce GTX 1080Ti)
-> Time per step reduced from 15.3s/step to 0.73s/step

@EParisot
Copy link

This is extremely usefull thankyou !

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