python中CNN多类图像分类的边界框预测

Question

我对4种类型的特定对象进行了训练和测试。我也有csv格式的装订框条件/关注区域坐标（x，y，w，h）。 该项目的主要目的是预测测试图像的类别以及感兴趣区域周围的边界框，并在图像上打印类别名称。

我已经基于keras库应用了CNN模型。它对测试集的给定图像进行分类。为了预测给定测试图像的边界框坐标，我应该改变什么？

from dataclasses import dataclass, field

Answer 1

您描述的任务是对象检测，通常需要更复杂的CNN模型。检查https://github.com/fizyr/keras-retinanet中是否有著名的神经网络架构之一。