tyagi-iiitv / PointPillars

GNU General Public License v3.0
105 stars 47 forks source link

Single Class Training prediction mismatch #41

Open user-2 opened 2 years ago

user-2 commented 2 years ago

Hi I made some changes to train for single class. I changed the final layer to sigmoid and changed config to only car and trained on KITTI dataset but there was a mismatch in the prediction script. I was getting Car class for my Ground Truths and 0 for my predictions. Any ideas where I could fix this?