torch>=1.12
torchvision
opencv-python>=4.5
numpy
tqdm
tensorboard