使用深度学习进行水印检测

数据挖掘 机器学习 神经网络 深度学习 美国有线电视新闻网 计算机视觉
2022-03-07 06:26:33

我有带和不带水印的图像。只有一种水印。我尝试过VGG16迁移学习,但结果很糟糕。

如果图像有水印,有哪些方法可以进行图像分类?

2个回答

您可以考虑将水印查找为对象检测问题,而不是图像分类问题。Faster R-CNN 和“你只看一次”(YOLO)都是通用的对象检测架构。

处理此类问题的一种好方法是使用地标检测。在这种情况下,您的问题看起来像是使用深度神经网络的分类和回归任务的组合。看看这里这里您应该指定在所有数据集中具有相同身份的点。然后您的网络将尝试找到每个点的位置。也看看这里