我正在寻找擅长计数对象的神经网络架构。例如，CNN 可以输出给定图像中球（或任何其他对象）的数量。我已经找到关于人群计数的文章，我正在寻找关于不同类型对象的文章。
感谢
使用 CNN 计算图像中的对象数量
人工智能
神经网络
深度学习
卷积神经网络
图像识别
2021-11-18 08:04:21
2个回答
如果您想使用神经网络计算对象的数量，您可以使用预训练的 YOLO 并移除底部预测层，并将特征馈送到分类前馈层，例如 1000 个类别，代表图像中的 0-999 个对象。然后你可以训练它并通过它传播梯度。例如，在 YOLO 的 pytorch 代码中，（来源：https ://github.com/eriklindernoren/PyTorch-YOLOv3 ）可以添加一个nn.Linear并使用交叉熵损失对图像数量进行分类。您还可以完全更改架构。也许你可以尝试添加层来重置或其他分类网络来计算对象的数量。希望这可以帮助你，并有一个美好的一天！
你可以运行 YOLO，然后统计出现的次数。