requirements.txt

# Base ----------------------------------------
matplotlib>=3.2.2
numpy>=1.21.6
opencv-python>=4.6.0
Pillow>=7.1.2
PyYAML>=5.3.1
requests>=2.23.0
scipy>=1.4.1
gradio>=3.36.1
tensorflow==2.12.0
tensorflow-datasets==4.9.2


# Plotting ------------------------------------
pandas>=1.1.4
seaborn>=0.11.0
gdown


# Extras --------------------------------------
psutil  # system utilization
thop>=0.1.1  # FLOPs computation