21 lines
1.0 KiB
Plaintext
21 lines
1.0 KiB
Plaintext
# pip install -r requirements.txt
|
|
Flask==3.1.1
|
|
waitress==3.0.2
|
|
PyYAML==6.0.2
|
|
|
|
ultralytics==8.3.203
|
|
# torch==2.7.0+cu126
|
|
# torchvision==0.22.0+cu126
|
|
# 注：torch版本包含CUDA，安装时需要特殊命令 pip install torch==2.7.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
|
|
tensorflow==2.20.0
|
|
|
|
opencv-python-headless==4.12.0.88
|
|
albumentations==1.0.0 # 把这个的版本改了就有好受的了
|
|
scikit-image==0.25.2
|
|
scikit-learn==1.7.2
|
|
pandas==2.3.1
|
|
# 可选
|
|
# pip install xformers==0.0.30
|
|
# triton==3.0
|
|
colorama==0.4.6
|
|
pillow==11.3.0 |