42 lines
1.2 KiB
Plaintext
42 lines
1.2 KiB
Plaintext
# 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 # 对应的这个依赖也需要降 |