torch
gradio
opencv-python
psutil
numpy
tqdm
pillow
torchvision
yolov5