问题描述:
Pytorch 模型转onnx模型过程中,onnx不支持派生类函数调用
在SSD类初始化里:
self.priorbox = prior_box.PriorBox(cfg[str(size)])
self.priors = Variable(self.priorbox.forward(),requires_grad=False)
在SSD类forward里:
解决方案:
在SSD类里去掉检测的操作,将检测操作独立出来一个函数去处理,如:
问题描述:
Pytorch 模型转onnx模型过程中,onnx不支持派生类函数调用
在SSD类初始化里:
self.priorbox = prior_box.PriorBox(cfg[str(size)])
self.priors = Variable(self.priorbox.forward(),requires_grad=False)
在SSD类forward里:
解决方案:
在SSD类里去掉检测的操作,将检测操作独立出来一个函数去处理,如: