License_Plate_Detection_Pytorch icon indicating copy to clipboard operation
License_Plate_Detection_Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon xuexingyu24

STN模型转onnx

Open hsl20130659 opened this issue 2 years ago • 2 comments

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

hsl20130659 avatar May 07 '22 15:05 hsl20130659

解决了吗

linnnff avatar Mar 13 '23 09:03 linnnff

@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?

jeungmin717 avatar Jun 16 '23 02:06 jeungmin717