GluonCV FasterRCNN to TVM

1

Hi everyone,

I’m trying to convert a pretrained GluonCV FasterRCNN to TVM Relay, but it seems that the NMS ops expect different inputs for the bbox coordinates.
The gluoncv one expects [score, xmin, ymin…] and is controlled through start_coords.
The tvm one expects [class, score, xmin…] but is not flexible (start_coord == 2).

Any hints on how to proceed?

Thank you

2

It’s solved in PR #2784: https://github.com/dmlc/tvm/pull/2784/files#diff-be20d04c8e0c8660a7f4f8cf76895b89L532