20 lines
349 B
Plaintext
20 lines
349 B
Plaintext
# if install failed, please change the pip source to 'https://mirror.baidu.com/pypi/simple'
|
|
|
|
# base deps
|
|
protobuf==3.20.3
|
|
|
|
# utils
|
|
psutil>=5.9.0
|
|
ruamel.yaml>=0.17.21
|
|
scipy>=1.9.3
|
|
tqdm>=4.64.1
|
|
opencv-python>=4.5.5.64
|
|
fast-histogram>=0.11
|
|
Pillow>=10.0.1
|
|
numpy<=1.26.4
|
|
|
|
# base
|
|
onnx==1.16.1
|
|
onnxoptimizer==0.3.8
|
|
onnxruntime>=1.16.0
|
|
torch>=1.13.1,<=2.2.0 |