how to convert the CondLane model to onnx or pt model (include net structure) ?

Turoad / lanedet

An open source lane detection toolbox based on PyTorch, including SCNN, RESA, UFLD, LaneATT, CondLane, etc.

Apache License 2.0

561 stars 93 forks source link

how to convert the CondLane model to onnx or pt model (include net structure) ? #52

Closed tangze5258 closed 2 years ago

tangze5258 commented 2 years ago

i need to covert the pth model to RKNN model to inference. The first step , the model must include net structure. so, i have tried to use torch.save or torch.jit.trace to save the whole net ,but errors occurred. anyone konw how to do that?

Turoad commented 2 years ago

Currently, condlanenet is not supported to export onnx. Many operators are not supported. If you want to, you need to write the custom operators for condlanenet.