Generalize Object Detection

Hi, How do i generalize object detection. i have tested Faster RCNN on new images of car and it detects the car but it also detects other parts of the car.

How do i generalize? How to make model detect only whole car and not its parts.
Would Image augmentation during training help me ?


Hi! Which data are you training on?

Hi, Thank you for your reply.

Data I am using is from Kaggle.

in my case non-maximum suppression helped me a lot.

I looked some images from the dataset. Are there images of tires or any other parts of cars? Or just images of cars and empty roads?

I don’t think so. I checked some of the images of dataset (not all of them) and there were only car images. there is not empty roads. all the images have at least 1 car in the image.

That might be another solution, adding empty road images without cars. That may reduce false positives of object detection of which Faster R-CNN is tend to.

I started very recently in the DL. Specifically in object detection I am starting my first projects so I have almost no experience.
I think the model takes into account the tires, the mirrors, etc. to identify the cars, and it is not very far from what we are looking for. If you could increase the set of images (it is the first thing I would try), then regulation techniques. I don’t think you can add new images since you are using a DB from Kaggle. Are you on a Kaggle competition or just downloading and using the images on your PC?

You can add same images but without bounding boxes maybe.
Yes, model takes into account other small parts of the car and that was a problem.
but non-maximum suppression is the solution for that.
No, i’m not on competition. Just downloading.

You can use some tools like CVAT, or Labelbox for YOLO. It depends on your models.

Check: GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Thanks. I’ll check it!

1 Like

Thanks. Yes, there are many tools for that.