【OpenCV】inpaintを使ってオブジェクト(矩形)を画像から消去するコード【Python】

【OpenCV】inpaintを使ってオブジェクト(矩形)を画像から消去するコード

【OpenCV】inpaintを使ってオブジェクト(矩形)を画像から消去するコード

物体検出(SSD)でキャベツ列を検出して出力された矩形（赤色の長方形）を消したいと思って調べたら、OpenCVのinpaint関数が使えるようだった。

学習用データ用に矩形のない生の画像データが欲しかったので試した。

コード【Python】

コードは↓
今回は動画から矩形を消すこととした。

#!/usr/bin/env python
# coding: utf-8

import numpy as np
import cv2 

cap = cv2.VideoCapture('cabbage.mp4')
fourcc = cv2.VideoWriter_fourcc('m','p','4','v')
out = cv2.VideoWriter('cabbage_inpainted.mp4', fourcc, 20.0, (int(cap.get(3)), int(cap.get(4))))

# マスクを作成（これは2値画像で、消したい部分のみを白色にする）
mask = np.zeros((int(cap.get(4)), int(cap.get(3))), dtype=np.uint8)
# (550,100)と(750,600)は赤色の矩形の頂点の座標
mask = cv2.rectangle(mask, (550,100), (750,600), color=(255,255,255),thickness=2)

while(cap.isOpened()):
    ret, frame = cap.read()
 
    if ret == True:

        frame = cv2.inpaint(frame, mask, 3, cv2.INPAINT_TELEA)
    
        cv2.imshow('inpainted', frame)
        out.write(frame)
    
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break
    
　　else:
        break

cap.release()
out.release()
cv2.destroyAllWindows()

これを実行すると結果は以下のようになった。