How to train vehicle recognition model

Question

Any pre-trained models that do vehicle recognition (Make, Model, Color)?, if not which is the best approach to train a model, Through Azure AI services or provision a computing instance that will train?

Answer 1

@Sihle Ndlovu Thanks for reaching out to us, there is currently no pre-trained model for vehicle recognition, you need to train your custom model.

There are a few options for custom your model, one is Azure Computer vision - Azure custom vision as below document - https://learn.microsoft.com/en-us/azure/ai-services/custom-vision-service/overview

Azure offers a Custom Vision service that supports object detection and image classification. While it doesn’t offer vehicle-specific models out of the box, you can use it with labeled images to train a model specific to your needs.

Moreover, you can also try ResNet50 or VGG16 pre-trained or YOLO but you still need to train your model.

There are some good open-source datasets for training a model to classify car make and model:

1. Stanford Cars Dataset: Contains 16,185 images of 196 classes of cars. It has less diversity but high image quality and standardized cropping.
2. [CompCars Dataset]: It contains data for both car classification and fine-grained recognition tasks. For classification, it has 431 car makes with 34,991 images. For fine-grained recognition, it has 13,858 images of 171 car models from the 431 makes.
3. VehicleID Dataset: A large dataset with 221,763 images of 26,267 vehicles captured by surveillance cameras. It has less standardized images but huge diversity.
4. Car Make and Model Recognition Dataset: Created from Google image search results, it contains 63,000 low-resolution images across 183 car makes and models. Helpful for evaluating real-world performance.

I hope this helps.

Regards,

Yutong

-Please kindly accept the answer if you feel helpful to support the community, thanks a lot.