埋坑埋坑埋坑埋坑
安装过程
cd trt-engine-explorer/
# 修改requirements.txt
pip install -U scikit-learn
pip install Werkzeug==2.2.2
pip install dtale==2.2.0 --no-deps
setuptools # for qgrid
wheel # for qgrid
# protobuf==3.16.0
# onnx==1.12.0
numpy
pandas==1.1.5
plotly
qgrid
graphviz
jupyter
netron
openpyxl # for excel reporting
ipywidgets==7.7.2
ipyfilechooser
jupyterlab
jupyter-dash
pytest
dtale
xlsxwriter
Flask==2.0.0
python utils/process_engine.py /code/lab/xxx.onnx /code/lab/temp minShapes=images:1x128x128x1 optShapes=images:1x1024x1024x1 maxShapes=images:1x1280x1280x1 shapes=images:1x1024x1024x1 fp16