Count objects found by the neural network

9 visualizzazioni (ultimi 30 giorni)
I'm working with neural networks and I need to quantify how many objects there are in an image through the neural network.
For example:
Want to make a classification that can tell me how many balls there are in this image and show which ones are balls.
net = googlenet;
I = imresize(imread('ball.jpg'), [224 224]);
classify(net, I)
Now I want to know how many balls are in this image.

Risposta accettata

Antoni Woss
Antoni Woss il 1 Feb 2022
A good place to start with detecting objects in images using a neural network is through using an object detector network architecture, such as a "You Only Look Once" (YOLO) network. See the documentation example for training such a network to detect vehicles in an image - https://www.mathworks.com/help/vision/ug/train-yolo-v2-network-for-vehicle-detection.html. If you want to count up the number of vehicles the network has detected, you can count the number of rows of the bboxes variable, i.e. size(bboxes,1).

Più risposte (0)

Categorie

Scopri di più su Deep Learning Toolbox in Help Center e File Exchange

Community Treasure Hunt

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

Start Hunting!

Translated by