| # 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,<=1.11.0 | |
| onnxruntime>=1.10.0 | |
| torch>=1.10.1,<=2.4.0 | |
| # tensorflow>=1.12.0,<=2.6.2 |