search

NVIDIA-ISAAC-ROS / isaac_ros_pose_estimation

Deep learned, NVIDIA-accelerated 3D object pose estimation
https://developer.nvidia.com/isaac-ros-gems
Apache License 2.0
141 stars 21 forks source link

How to generate custom training data for centerpose #29

Open ArghyaChatterjee opened 11 months ago

ArghyaChatterjee commented 11 months ago

Hello, I am trying to generate dataset for training centerpose for custom models. How can I do that ?? Is there any guideline to do that ??

hemalshahNV commented 10 months ago

TAO will be releasing with support for training Centerpose soon soon including generating synthetic datasets from Isaac Sim.

kajananchinniahNV commented 10 months ago

Adding on, for real data, the original repository has a data section which discusses how they pre-processed the Objectron dataset for CenterPose. This can likely be adapted for your use-case, as long as you can make your custom data follow the Objectron dataset's format.

The upcoming TAO release will also contain an example that shows downloading the Objectron dataset and processing it for use in CenterPose.

monajalal commented 7 months ago

@kajananchinniahNV @hemalshahNV Is there any update on having a Writer for CenterPose in Replicator of Isaac Sim? Like we have Kitti and DOPE writer. Or is there a script that authors could share for converting BasicWriter to CenterPose?

rubenponsaers commented 5 months ago

Any update on how to train a model on a custom dataset using BundleSDF?