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