43 lines
1.1 KiB
Plaintext
43 lines
1.1 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==2.0.2
|
|
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 --extra-index-url https://download.pytorch.org/whl/cu121
|
|
torchvision --extra-index-url https://download.pytorch.org/whl/cu121
|
|
torchaudio --extra-index-url https://download.pytorch.org/whl/cu121
|
|
tqdm>=4.66.3
|
|
ultralytics>=8.2.34 # https://ultralytics.com
|
|
|
|
# Plotting --------------------------------------------------------------------
|
|
pandas==2.2.3
|
|
seaborn>=0.11.0 |