search

V2AI / Det3D

World's first general purpose 3D object detection codebse.
https://arxiv.org/abs/1908.09492
Apache License 2.0
1.48k stars 299 forks source link

How to train model for a custom object ? #139

Closed pra-dan closed 3 years ago

pra-dan commented 3 years ago

I am planning to use Det3D to train a detector for a custom object (a toy or some table). I prefer using only RGB. How should I go about it ? Any starting notes ? Thanks in advance :)

poodarchu commented 3 years ago

Add your datasets by inherit the base Dataset class : https://github.com/poodarchu/Det3D/blob/master/det3d/datasets/custom.py

And specify your image read func and transforms in getitem and configs.