pre-train with new datasets

[CongDuy038]

hi, now i have created a linemod-like dataset and i want to pretrain. So how to do that? Thanks @liuyuan-pal

[liuyuan-pal]

Hi, do you mean that you wanna train the Gen6D on your dataset? In this case, you may follow https://github.com/liuyuan-pal/Gen6D#training and the training is relatively complex because it contains three stages, I suggest you only fine-tune one stage. In most cases, detector and refiner are more critical for better performance.

[CongDuy038]

@liuyuan-pal thanks. can you explain to me about datasets co3d, coco, shapenet and google_scanned? Can I fine-tune this datasets during pre-training?

[liuyuan-pal]

You only need to finetune it on your own datasets but not these provided datasets.