23 lines
398 B
Plaintext
23 lines
398 B
Plaintext
# if install failed, please change the pip source to 'https://mirror.baidu.com/pypi/simple'
|
|
|
|
# base deps
|
|
numpy==1.23.4
|
|
protobuf==3.20.3
|
|
flatbuffers==2.0
|
|
|
|
# utils
|
|
requests==2.28.1
|
|
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
|
|
|
|
# base
|
|
onnx==1.13.1
|
|
onnxoptimizer==0.3.8
|
|
onnxruntime==1.14.1
|
|
torch==1.13.1
|
|
torchvision==0.14.1
|
|
tensorflow==2.8.0 |