Classify bounding boxes using Neural Net, in a video.

2 visualizaciones (últimos 30 días)
Rana Raees Ahmed Khan
Rana Raees Ahmed Khan el 11 de En. de 2018
Hi, I am working on processing a video to detect and classify cars according to their model. My aim is to draw bounding boxes around detected cars, and classify those cars using neural net, and display the model of the car as its label above the bounding box. what i have achieved so far is:
  • draw bounding boxes on detected cars,
  • get all the bounding boxes and classify them using a .mat file, which is a trained neural network for car models.
  • display label on the bounding boxes of each car.*
my problem is:
  • either i get only one label applied on all bounding boxes of cars, on every frame.
  • or i get no label at all.
what i want is:
  • to classify all bounding boxes according to their models in the video.
  • display the label on the bounding box after classification, on the video.
please see my attached code. m-file. any help would be highly appreciated.
this is the output i get, where all bounding boxes have only one label: "CIVIC".
p.s I used alexNet and did transfer learning to train the network for vehicle make and model recognition.

Respuestas (0)

Categorías

Más información sobre Recognition, Object Detection, and Semantic Segmentation en Help Center y File Exchange.

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by