How to create my own dataset to train a model?

stefan-ainetter / grasp_det_seg_cnn

Code for ICRA21 paper "End-to-end Trainable Deep Neural Network for Robotic Grasp Detection and Semantic Segmentation from RGB".

BSD 3-Clause "New" or "Revised" License

133 stars 17 forks source link

How to create my own dataset to train a model? #33

Open elenacliu opened 11 months ago

elenacliu commented 11 months ago

I want to train the model on GIGA dataset, which is represented in 3d object combination. To get a dataset from GIGA which can be used for this repo, I need to get 2d label instance mask. How do you generate that mask?