how to make custom data sets?

zubair-irshad / CenterSnap

Pytorch code for ICRA'22 paper: "Single-Shot Multi-Object 3D Shape Reconstruction and Categorical 6D Pose and Size Estimation"

https://zubair-irshad.github.io/projects/CenterSnap.html

286 stars 47 forks source link

how to make custom data sets? #32

Open nahyun93 opened 7 months ago

nahyun93 commented 7 months ago

Hello, I wanted to train the Centersnap model by establishing a custom data set. I have RGB, depth images, and 3D object files of the objects to be detected. Can I make a dataset using only these? How can I get mask.png, coordinate.png, meta.txt, and files in the gts and obj_models folders?