- Linux
- Python 3.5+
- PyTorch 1.11
- TensorBoard
- CUDA 11.0+
- GCC 4.9+
- NumPy 1.11+
- PyYaml
Part of NMS is implemented in C++. The code can be compiled by
cd ./libs/utils
python setup_nms.py install --user
cd ../..The code should be recompiled every time you update PyTorch.