search

xuexingyu24 / License_Plate_Detection_Pytorch

A two stage lightweight and high performance license plate recognition in MTCNN and LPRNet
Other
641 stars 171 forks source link

STN模型转onnx #42

Open hsl20130659 opened 2 years ago

hsl20130659 commented 2 years ago

大佬，STN模型转onnx貌似转不了

linnnff commented 1 year ago

解决了吗

jeungmin717 commented 1 year ago

@linnnff @hsl20130659

Hi, How did you convert stn model to onnx? I am so much struggling with stn. stn has pytorch function which is not supported converting opset <16

grid = F.affine_grid(theta, x.size()) x = F.grid_sample(x, grid)

these two functions are not supported. Did you figure it out?