使用Tensorflow来对图片相似度进行比较时,发现在进行相同的一张图片之间的比较时,是可以比较得出结果的。即MSE=0,ssim=1。说明这两张图片是相同的。其中主要MSE和SSIM公式分别通过下面的公式获得。
均分误差:
结构相似性:
但是在进行两个不同图片之间的比较时就会出现下面的报错:ValueError: operands could not be broadcast together with shapes (182,176) (185,173)。想了半天才明白应该是两个图片的长宽不一样,那么处理起来就会报错啊。然后我就进行了图片的裁剪处理,其主要代码如下:
def crop_or_pad(filePath):
filename = [filePath]
filename_queue = tf.train.string_input_producer(filename)
reader = tf.WholeFileReader()
key, value = reader.read(filename_queue)
images &#