自己做的数据集,是点标注(每个细胞中心),一共有四类细胞,要做细胞检测分类,从而得到细胞的具体位置,从而做下游分析。
问题是现在大多数目标检测模型都是box标注,无法使用,最多只能是将点标注转换成box标注,但是因为细胞大小不一致,会有精度的下降,只能用这种方法跑跑对比实验。
我应该尝试去运行什么模型来使用我的数据集?或者有什么效果比较好的使用点标注的模型?
深度学习图像处理目标检测图像分割计算机视觉 06--图像分类
可以参考下
尝试使用One-stage目标检测模型、CenterNet模型或RetinaNet模型来处理您的点标注数据集,以实现细胞检测分类任务
PointNet是由斯坦福大学提出的一个深度学习网络,专门设计用于处理点云数据的。或者,CenterNet是一个用于2D目标检测的模型,它通过预测目标的中心点和类别。虽然它不是专门为点标注设计的,但是你可以尝试将其应用于你的数据,因为它对大小不一的目标具有较好的适应性。
YOLOv5也可以用于2D目标检测,并且可以处理大小不一的目标。
【以下回答由 GPT 生成】
对于这个问题,可以尝试使用点标注数据集进行训练的模型,并尝试使用一种方法来解决细胞大小不一致的问题。一种可行的方案是使用基于关键点(keypoint)的目标检测模型,如Hourglass网络。下面是具体的解决方案:
标签文件应包含每个细胞的中心点坐标。
Hourglass网络模型:
这个模型通过多个阶段的卷积和下采样、上采样操作来预测图像中关键点的热图。
训练模型:
在训练过程中,将中心点标注转换为关键点形式,即将每个细胞的中心点作为一个关键点来训练模型。
测试模型:
使用模型对测试集中的图像进行推断,得到每个细胞的关键点位置。
解决细胞大小不一致问题:
总结: Hourglass网络是一种适合使用点标注数据集进行训练的目标检测模型,可以用于细胞检测分类并获取每个细胞的具体位置。通过数据扩充和关键点分布估计等方法,可以一定程度上解决细胞大小不一致带来的准确性下降问题。
【相关推荐】