onnx2torch icon indicating copy to clipboard operation
onnx2torch copied to clipboard

Unable to successfully convert retinanet R50

Open NTUTshaofong opened this issue 1 year ago • 1 comments

Traceback (most recent call last): File "/home/li/testcode/tool2/convert_dataloader.py", line 490, in fp_model = copy.deepcopy(convert(onnx_model).to(device))

File "/home/li/.local/lib/python3.6/site-packages/onnx2torch/converter.py", line 107, in convert version=version, File "/home/li/.local/lib/python3.6/site-packages/onnx2torch/node_converters/registry.py", line 69, in get_converter raise NotImplementedError(f'Converter is not implemented ({description})') NotImplementedError: Converter is not implemented (OperationDescription(domain='', operation_type='NonZero', version=13))

I tried changing the opset but apparently it doesn't support non-zero operators. Can you tell me the details of the method the authors originally tested ? thanks

NTUTshaofong avatar Nov 06 '23 08:11 NTUTshaofong

Hi @NTUTshaofong, we test convertion on the model retinanet_r50_fpn from the MMDetection framework. You can find it in our tests. You can also download our ONNX by the following link.

Can you please provide the details about your checkpoint?

andreysher avatar Nov 08 '23 10:11 andreysher