Skip to content

Request: Support for deform_conv2d from an alternate repository to enable better ONNX export with dynamic_shape #127

@alvarofsan

Description

@alvarofsan

Hi,

I was wondering if it would be possible to modify the current implementation of the model to use the DeformConv2dFunction function from MMCV instead of the official implementation from Torchvision.

The reason for this request is that the developers of MMCV have also implemented support for exporting this operator to ONNX, along with its associated ONNX Runtime operator. This would enable exporting the model to the ONNX format without relying on the current solution, deform_conv2d_onnx_exporter, which is quite slow, and would help address potential issues related to dynamic_shape handling.

Would you consider integrating or optionally supporting this alternative implementation? If needed, I’d be happy to contribute. I understand that a change of this type might require retraining the network.

Thank you for your time and consideration!

Best regards,
Álvaro.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions