Blanktar

  1. top
  2. blog
  3. 2015
  4. 08

python/OpenCVで複数の画像から背景だけを取り出す

同じ場所で撮られた複数枚の画像、あるいは動画から背景だけを抽出する、というのをしばらく前にやったのでメモ。 新しいバージョンのOpenCVだと組み込みで背景抽出の関数があるそうですが、これはやっつけ仕事でやったやつなので、シンプルに動体検出の応用で出来ています。

ここにサイコロの画像が6枚あります。

キーボードの上にサイコロが転がっている画像6枚

この画像から背景だけ、つまりはキーボードだけが写った画像を作ろうというのが今回の目的。

とりあえずソースコードから。

import numpy
import cv2


def guessBackground(imgs, threshold=16):
    old = imgs[0]
    result = numpy.zeros_like(imgs[0])

    for img in imgs[1:]:
        mask = cv2.absdiff(old, img)
        result = result*(mask >= threshold) + img*(mask < threshold)

        old = img

    return result


imgs = []
for i in range(6):
    imgs.append(cv2.imread('{0}.jpg'.format(i)))

result = guessBackground(imgs)
cv2.imshow('result', result))

cv2.waitKey()

cv2.absdiffを使って二枚の画像の差分を計算して、差の少ない部分のデータだけを蓄積していくだけのコードです。とってもシンプル。

このコードで生成した出力がこちら。

背景だけを抽出してサイコロが無くなったっぽい画像

ほこりっぽい。掃除しよ…。というのはともかくとして、わりと綺麗に消えています。 エンターキーのあたりに影が残っているのは、入力画像のうち3枚がエンターにサイコロが乗っているから、かな。当然ですが入力が偏ればこんな結果になります。

入力画像を入力された順番でしか比較しないので微妙な精度ですが、順番を入れ換えたりすればもっと綺麗に出来たりもします。自動で上手いことやるのも可能なはず。 まあでも、これだけ適当でもそれっぽいのが出来るよ、ということで。