Count objects found by the neural network

Question

Guilherme Franklin 2022-2-1

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1640260-count-objects-found-by-the-neural-network

回答： Antoni Woss 2022-2-1

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.

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Antoni Woss 2022-2-1

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1640260-count-objects-found-by-the-neural-network#answer_886550

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).