Home icon
Build and train computer vision models to detect car positions in images using Amazon SageMaker and Amazon Rekognition

Blog



This article demonstrates how to build a computer vision solution for detecting car positions in images using two approaches: Amazon Rekognition and a custom SageMaker model.

  • Amazon Rekognition detects car wheels, then infers orientation using rule-based postprocessing
  • Custom Detectron2 model trained on SageMaker for more granular car part detection
  • Solution uses Lambda functions behind API Gateway for serverless deployment
  • Detectron2 fine-tuned on car parts segmentation dataset with ~500 annotated images
  • Web application built with AWS Amplify allows users to upload images and select detection method
  • Trained model hosted on SageMaker real-time endpoint using GPU instance
  • Rekognition Custom Labels option available for improved wheel detection accuracy
  • Complete infrastructure provisioned using AWS CDK for easy deployment

The article provides a flexible, serverless architecture for car pose detection that can be adapted for other object orientation use cases, with choice between managed service or custom ML model approaches.



Go to article

The AWS News Feed is currently looking for gold sponsors. If you want to support the AWS community and reach a large audience of AWS professionals, consider sponsoring the AWS News Feed.

Related articles

The AWS News Feed is currently looking for silver sponsors. If you want to support the AWS community and reach a large audience of AWS professionals, consider sponsoring the AWS News Feed.