pytorch>=1.5.0
torchvision>=0.5.0
tensorboardX>=2.0opencv-python
tqdm

[tensorflow]
tensorflow
