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

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.

Geek Repo

Github PK Tool

imageai yolov3 resnet-50 classification keras tensorflow transferlearning deeplearning vehicle make model object detection

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.

imageai yolov3 resnet-50 classification keras tensorflow transferlearning deeplearning vehicle make model object detection

Languages

Language:Jupyter Notebook 100.0%