Hello,
I'm following your repositories and I have already done training with modifying the code in common.py . Now, I would like to export the model on onnx and you suggested to remove the detect layer from the graph before exporting it.
I would like the ask that where exactly I have to modify model.model[-1].export = True in export.py.
Thank you in advance for your help.
Hello,
I'm following your repositories and I have already done training with modifying the code in common.py . Now, I would like to export the model on onnx and you suggested to remove the detect layer from the graph before exporting it.
I would like the ask that where exactly I have to modify model.model[-1].export = True in export.py.
Thank you in advance for your help.