torch >= 1.4.0
torchvision >= 0.5.0
pyyaml
opencv-python
numpy
lmdb
tqdm
tensorboard >= 1.15.0
ninja