export_model.py - list_of_lines[165] = " [1344, 1344], 1344 \n"

[kaustubhsapru]

📚 Documentation Issue

This issue category is for problems about existing documentation, not for asking how-to questions.

• Provide a link to an existing documentation/comment/tutorial:

• https://colab.research.google.com/drive/1ZFdkdIAjD0ldhJ9TEhzTL1bndLyqm2Rd?usp=sharing#scrollTo=fDKNGh6ZW4dD

• How should the above documentation/comment/tutorial improve:
  For this snippet :

a_file = open("/content/detectron2/tools/deploy/export_model.py", "r")
list_of_lines = a_file.readlines()
list_of_lines[165] = " [1344, 1344], 1344 \n"

a_file = open("/content/detectron2/tools/deploy/export_model.py", "w")
a_file.writelines(list_of_lines)
a_file.close()
exit(1)

line 165 in export_model.py reads : "# get a first batch from dataset"
What has to change is this code for an error free conversion from .yaml to onnx ?

[Huxwell]

I'm not a maintainer, but what is the actual error you are getting?
You can manually change the relevant line in export_model.py as well.