20 lines
359 B
Plaintext
20 lines
359 B
Plaintext
# if install failed, please change the pip source to 'https://pypi.tuna.tsinghua.edu.cn/simple/'
|
|
|
|
# base deps
|
|
protobuf>=3.12.2,<=3.19.6
|
|
|
|
# utils
|
|
psutil>=5.9.0
|
|
ruamel.yaml>=0.17.4
|
|
scipy>=1.5.4
|
|
tqdm>=4.64.0
|
|
opencv-python>=4.5.5.64,<=4.6.0.66
|
|
fast-histogram>=0.11
|
|
numpy<=1.26.4
|
|
|
|
# base
|
|
onnx==1.10.0
|
|
onnxoptimizer==0.2.7
|
|
onnxruntime==1.10.0
|
|
torch>=1.10.2,<=2.1.0
|