| # if install failed, please change the pip source to 'https://pypi.tuna.tsinghua.edu.cn/simple/' | |
| # base deps | |
| protobuf>=4.21.0,<=4.25.4 | |
| # 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 | |
| numpy<=1.26.4 | |
| # base | |
| onnx==1.16.1 | |
| onnxruntime==1.17.0 | |
| torch>=1.13.1,<=2.2.0 | |
| # tensorflow<=2.16.0rc0 |