TRI-ML / dd3d

Official PyTorch implementation of DD3D: Is Pseudo-Lidar needed for Monocular 3D Object detection? (ICCV 2021), Dennis Park*, Rares Ambrus*, Vitor Guizilini, Jie Li, and Adrien Gaidon.
MIT License
451 stars 74 forks source link

where to change the train class num 5 to 3 #38

Closed 694376965 closed 1 year ago

694376965 commented 1 year ago

Hi, I find that the dd3d is trained with 5 classes: Car,Pedestrian,Cyclist,Van,Truck , But I just want to train 3 classes (Car,Pedestrian,Cyclist). where to modify the code to train 3 classes ? I am looking forward to your reply! Thank you very much!