我想检测图像中的缺陷/异常。由于缺乏异常图像,我尝试以无人监督的方式解决问题。到目前为止,我训练了一个变分自动编码器和一个带有“好”图像的生成对抗网络。现在,我有一个编码器/解码器网络,它能够生成没有异常的图像。这里有两个例子:
原图 | 编码器的裁剪图像和输入| 解码器的输出
在这些图像中,如果对象是旋转的(不是垂直的),那么它就是异常(如第二张图像)。为了测量编码器/解码器网络的输入/输出之间的差异,我尝试了结构相似度度量 SSIM。然而,结果并不令人满意,因为许多没有异常的图像也具有低相似度值。此外,像素方面的 MAE 或 MSE 指标也不好,因为它们看起来并不特别关注结构并且对亮度非常敏感。这是 SSIM 指标的结果(绿色是很好的例子,红色是异常):
我要避免的是为检测添加特定规则,在此示例中检测垂直线,因为我想检测许多其他具有不同异常的对象。
你对另一个图像相似度指标有什么建议吗?我是在正确的道路上还是有其他/更好的方法来进行无监督异常检测?
PS:只有少数图像有异常(<10 张图像),但很多没有异常。