isaac_ros_pose_estimation icon indicating copy to clipboard operation
isaac_ros_pose_estimation copied to clipboard

How to generate custom training data for centerpose

Open ArghyaChatterjee opened this issue 1 year ago • 4 comments

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 ??

ArghyaChatterjee avatar Oct 20 '23 18:10 ArghyaChatterjee

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

hemalshahNV avatar Oct 23 '23 22:10 hemalshahNV

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.

kajananchinniahNV avatar Oct 23 '23 23:10 kajananchinniahNV

@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?

monajalal avatar Feb 13 '24 15:02 monajalal

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

rubenponsaers avatar Mar 30 '24 10:03 rubenponsaers