traveller59 / second.pytorch

SECOND for KITTI/NuScenes object detection
MIT License
1.73k stars 722 forks source link

Need tips on improving performance on custom dataset #489

Open Allamrahul opened 1 year ago

Allamrahul commented 1 year ago

Hi, I am using this repo, specifically PointPillars for 3D object detection on custom point cloud dataset acquired using Ouster LIDAR. I am able to train the network and able to evaluate. My object of interest are farm posts in a barn. I used the ped_cycle xyres_16.config as reference since posts have a similar aspect ratio to a pedestrian and adapted it slightly. My objective is to detect four farm posts out of the many that are in front of the vehicle.

Currently, I have an eval of 30 point clouds. In that, I am able to get the detections in 75% of them. But in the others, its missing the posts. Not sure why that is. I have tried

BUT I am unable to achieve a performance greater than 75%.

The bev I is around 0.4. I need high recall i.e. getting all the 4 posts.

I would really like some suggestions in terms of what I can tweak in the config for better performance. I am attaching the config for reference.

xyres16.txt