wozwdaqian / TensorRT-DAB-DETR

发际线与我作队
Apache License 2.0
3 stars 1 forks source link

Missed onnxsim package #1

Open haohaibo opened 2 years ago

haohaibo commented 2 years ago
root@5bf542c3e980:/workspace/TensorRT-DAB-DETR# python export_onnx_sim.py
Traceback (most recent call last):
  File "export_onnx_sim.py", line 25, in <module>
    from onnxsim import simplify
ModuleNotFoundError: No module named 'onnxsim'
root@5bf542c3e980:/workspace/TensorRT-DAB-DETR# pip install onnxsim
Looking in indexes: https://pypi.org/simple, https://pypi.ngc.nvidia.com
ERROR: Could not find a version that satisfies the requirement onnxsim (from versions: none)
ERROR: No matching distribution found for onnxsim
root@5bf542c3e980:/workspace/TensorRT-DAB-DETR# ls
'=0.6.0'       Makefile.inc   engine.py                       main.py       nsys2.png          resources
'=1.5.0'       README.md      export_onnx.py                  markdown.md   nsys3.png          run_with_submitit.py
 COCODIR.zip   ToNetron.py    export_onnx_sim.py              model_zoo     nsys4.png          test.py
 LICENSE       coco.sh        figure                          models        oneflow_LN         testLayerNormPlugin.py
 LN_onnx       datasets       inference_and_visualize.ipynb   nsys1.png     requirements.txt   util
root@5bf542c3e980:/workspace/TensorRT-DAB-DETR#
root@5bf542c3e980:/workspace/TensorRT-DAB-DETR# vim export_onnx_sim.py
root@5bf542c3e980:/workspace/TensorRT-DAB-DETR# pip install onnx-simplifier
Looking in indexes: https://pypi.org/simple, https://pypi.ngc.nvidia.com
wozwdaqian commented 2 years ago

pip install -i https://pypi.douban.com/simple -U onnx-simplifier --user

Zaughter commented 11 months ago

hi friend. So have you finally successfully export it to onnx? I met a few warning and maybe need your help.