My configurations: Ubuntu 18.04, Python 3.6.9, PyTorch 1.3.1, CUDA 10.1, CUDNN 7.6.5, numpy 1.19.2, opencv 4.4.0, matplotlib 3.3.1.
You should also install the library RoiAlign.
Download images and models from Dropbox or Baidu disk with endorsement code jh7g, then run
python run.py
All the detection results are in the "results" directory.
The entire document structure: