Count objects found by the neural network

4 次查看(过去 30 天)
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.

采纳的回答

Antoni Woss
Antoni Woss 2022-2-1
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).

更多回答(0 个)

类别

Help CenterFile Exchange 中查找有关 Deep Learning Toolbox 的更多信息

Community Treasure Hunt

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

Start Hunting!

Translated by