Structural Similarity (SSIM) - 结构相似性帧差分

Structural Similarity (SSIM) - 结构相似性帧差分#

# 结构相似性(SSIM)示例
#
# 此示例展示了如何在CanMV Cam上使用SSIM算法来检测两个图像之间的差异。 
# SSIM算法比较两个图像之间的8×8像素块以确定两个图像之间的相似性得分。

import sensor, image, os, time

# 如果sim.min() 低于此值,则图像可能已更改。
MIN_TRIGGER_THRESHOLD = -0.4

sensor.reset()                      # 复位并初始化摄像头
sensor.set_pixformat(sensor.RGB565) # 设置摄像头输出格式为 RGB565(也可以是GRAYSCALE)
sensor.set_framesize(sensor.QVGA)   # 设置摄像头输出大小为 QVGA (320x240)
sensor.skip_frames(time = 2000)     # 跳过2000帧
sensor.set_auto_whitebal(False)     # 关掉自动白平衡
clock = time.clock()                # 创建一个clock对象,用来计算帧率

# 从主帧缓冲区的RAM中取出以分配第二帧缓冲区。
# 帧缓冲区中的RAM比MicroPython堆中的RAM多得多。
# 但是,在执行此操作后,您的某些算法的RAM会少得多......
# 所以,请注意现在摆脱RAM问题要容易得多。
# 然而,帧差分不会占用帧缓冲区中的大量额外空间。
# 但是,如果你这样做,像AprilTags这样的东西会起作用,也可能不会起作用...
extra_fb = image.Image(size=(sensor.width(), sensor.height()))

print("About to save background image...")
sensor.skip_frames(time = 2000) # Give the user time to get ready.
# extra_fb.replace(sensor.snapshot())
extra_fb.draw_image(sensor.snapshot(), 0, 0)
print("Saved background image!")

while(True):
    clock.tick()                # 更新计算帧率的clock
    img = sensor.snapshot()     # 拍照,获取一张图像
    sim = img.get_similarity(extra_fb)
    change = "- Change -" if sim.min() < MIN_TRIGGER_THRESHOLD else "- No Change -"

    print(clock.fps(), change, sim)

具体接口定义请参考 get_similarity