Skip to content

centerpose_converter.py is missing for converting pth to onnx inside ros2 docker #67

@monajalal

Description

@monajalal

Similiar to dope_converter.py, could you please share the script for centerpose_converter.py for converting the pth model trained using TAO toolkit to ONNX model inside the ros2 docker?

From isaac_ros_dope tutorial:

Image

ros2 run isaac_ros_dope dope_converter.py --format onnx \ --input ${ISAAC_ROS_WS}/isaac_ros_assets/models/dope/Ketchup.pth --output ${ISAAC_ROS_WS}/isaac_ros_assets/models/dope/Ketchup.onnx --row 720 --col 1280

https://nvidia-isaac-ros.github.io/repositories_and_packages/isaac_ros_pose_estimation/isaac_ros_dope/index.html#quickstart

Currently the tutorial for isaac_ros_centerpose directly downloads the onnx from ngc website. I need to be able to reproduce the results for my custom object class.

Probably same or slight modification of https://github.com/NVIDIA/tao_pytorch_backend/blob/main/nvidia_tao_pytorch/cv/centerpose/scripts/export.py

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