Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[Feature] Support ONNX and TensorRT exportation of RTMO models #2597

Merged
merged 10 commits into from
Dec 14, 2023
Merged

[Feature] Support ONNX and TensorRT exportation of RTMO models #2597

merged 10 commits into from
Dec 14, 2023

Commits on Dec 11, 2023

  1. support ONNX&TensorRT exportation of RTMO

    @Ben-Louis
    Ben-Louis committed Dec 11, 2023
    Configuration menu
    Copy the full SHA
    fe7aef1 View commit details
    Browse the repository at this point in the history

  2. add configs for rtmo

    @Ben-Louis
    Ben-Louis committed Dec 11, 2023
    Configuration menu
    Copy the full SHA
    6ac1102 View commit details
    Browse the repository at this point in the history

  3. replace bbox expansion factor with parameter bbox_padding

    @Ben-Louis
    Ben-Louis committed Dec 11, 2023
    Configuration menu
    Copy the full SHA
    21dc0ba View commit details
    Browse the repository at this point in the history

  4. refine code

    @Ben-Louis
    Ben-Louis committed Dec 11, 2023
    Configuration menu
    Copy the full SHA
    e669d16 View commit details
    Browse the repository at this point in the history

  5. refine comment

    @Ben-Louis
    Ben-Louis committed Dec 11, 2023
    Configuration menu
    Copy the full SHA
    c3ec981 View commit details
    Browse the repository at this point in the history

Commits on Dec 13, 2023

  1. apply model.switch_to_deploy in BaseTask.build_pytorch_model

    @Ben-Louis
    Ben-Louis committed Dec 13, 2023
    Configuration menu
    Copy the full SHA
    2207861 View commit details
    Browse the repository at this point in the history

  2. fix lint

    @Ben-Louis
    Ben-Louis committed Dec 13, 2023
    Configuration menu
    Copy the full SHA
    97231f8 View commit details
    Browse the repository at this point in the history

Commits on Dec 14, 2023

  1. add rtmo into regression test

    @Ben-Louis
    Ben-Louis committed Dec 14, 2023
    Configuration menu
    Copy the full SHA
    e0d7166 View commit details
    Browse the repository at this point in the history

  2. add rtmo with trt backend into regression test

    @Ben-Louis
    Ben-Louis committed Dec 14, 2023
    Configuration menu
    Copy the full SHA
    8080b4d View commit details
    Browse the repository at this point in the history

  3. add rtmo into supported model list

    @Ben-Louis
    Ben-Louis committed Dec 14, 2023
    Configuration menu
    Copy the full SHA
    ce2658e View commit details
    Browse the repository at this point in the history