This is a tensorflow version of GHMC-Loss based on this paper Gradient Harmonized Single-stage Detector and the original pytorch version: https://github.com/libuyu/GHM_Detection.
This code uses the broadcast mechanism for parallel computing for all bins.
This code is implemented with
- Python 3.6
- PyTorch 0.4.1
- Tensorflow 1.11.0
If you find any bug in the code, please let me know.