# aicheck_v2 requirements
# Usage: pip install -r requirements.txt

# API -------------------------------------------------------------------------
fastapi~=0.74.1 # fastapi基础包
pydantic~=1.9.0 # pydantic
email-validator==1.1.3
PyJWT~=2.3.0
python-multipart==0.0.5
redis~=4.1.4
SQLAlchemy~=2.0.34
uvicorn~=0.17.5
uvicorn[standard]
loguru~=0.6.0
xlrd~=2.0.1
bcrypt==3.2.0
pymysql==1.0.2
pynvml==12.0.0
requests-toolbelt==1.0.0
python-socketio == 5.12.1


# YOLOV5 ----------------------------------------------------------------------
# BASE ------------------------------------------------------------------------
gitpython>=3.1.30
matplotlib>=3.3
numpy==1.24.0 # cuda版本的numpy需要降一下版本
opencv-python>=4.1.1
pillow>=10.3.0
psutil  # system resources
PyYAML>=5.3.1
requests>=2.32.2
scipy==1.13.1
thop>=0.1.1  # FLOPs computation
torch==2.3.1+cu121 --extra-index-url https://mirrors.aliyun.com/pytorch-wheels/cu121/
torchvision==0.18.1+cu121 --extra-index-url https://mirrors.aliyun.com/pytorch-wheels/cu121/
tqdm>=4.66.3
ultralytics==8.3.75  # https://ultralytics.com

# Plotting --------------------------------------------------------------------
pandas==2.2.3
seaborn==0.11.0 # 对应的这个依赖也需要降