Skip to content

How to export the model as ONNX #11

@LeonardoRez

Description

@LeonardoRez

Hello,

I'm interested in testing the HRNetV2-OCR+PSA(p) with OpenCV, but I believe there is no way to load a .pth file. I'm checking how to use pytorch to load the network and then export it to ONNX, but It seems that I need to know the model definition to properly load the network. Am I going in the right direction? Sorry if the question seems naive, I don't have a lot of experience with ML, especially with pytorch.

Thanks in advance

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions