| # if install failed, please change the pip source to 'https://pypi.tuna.tsinghua.edu.cn/simple/' | |
| # base deps | |
| protobuf>=4.21.6,<=4.25.4 | |
| # 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.10.0 | |
| onnxruntime>=1.10.0 | |
| torch>=1.10.1,<=2.4.0 | |
| # tensorflow>=2.5.0,<=2.14.0 |