我需要一种方法来判断一张图片是否是照片。我有一堆随机的图像文件(纸质文件扫描件、标志,当然还有相机拍摄的照片),我需要筛选出其中的照片以创建预览。
在Determine if image is photograph or drawing, quickly提出的解决方案只能在有限的情况下工作(例如,一些标志完全是黑色的带有白色字体,一些标志只有颜色,没有白色区域),有时候我会得到一张白纸的扫描图,里面包含多个照片和周围的空白区域 – 我也需要识别这些,因为然后我需要去除白色部分,并将扫描图上的照片保存为单独的文件。
回答:
你完成这个任务的过程可能应该类似于以下步骤:
-
从图像中提取特征(像素值、像素组、HoG、SIFT、GIST、DCT、小波、字典学习系数等,具体取决于你有多少时间)
-
以某种方式聚合这些特征,以便你得到一个固定长度的向量(直方图、金字塔方案)
-
应用标准分类(SVM、k-NN、神经网络、随机森林)或聚类算法(k-means、GMM等),并测量其效果(F1分数通常可以接受,对于两类问题,ROC可能更好)
-
如果对步骤3的结果不满意,从步骤1开始重复使用不同的特征
你提到的解决方案在步骤1和2方面似乎是相当合理的。
提取和聚合特征的下一个简单步骤可以是从图像中的所有像素值创建直方图。如果你有大量标记数据,你应该将这些特征输入到标准分类器中。否则,在这些直方图特征上运行聚类算法,并检查聚类分配是否与照片/非照片分配相关。