Skip to content

Object Detection via pre-trained YOLOv3 is used to detect vehicles from an image. Transfer learning on ResNet-50 outer layers with a two-class dataset trains for 50 epochs. 81% accuracy.

cchighman/ImageAI-YOLOv3-Fine-Tuning-Vehicle-Classification

Repository files navigation

ImageAI-YOLOv3-Fine-Tuning-ResNet-Vehicle-Make-Classification

81% accuracy after transfer learning. Focused on classifying Toyota Prius models.

About

Object Detection via pre-trained YOLOv3 is used to detect vehicles from an image. Transfer learning on ResNet-50 outer layers with a two-class dataset trains for 50 epochs. 81% accuracy.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published