torch>=1.5.0
torchvision>=0.6.0
opencv-python
tqdm

[tensorflow]
tensorflow
