Repair.py

# -*- coding: UTF-8 -*-
import numpy as np
from PIL import Image
import os
import cv2


def repair(frame):
    # 把 BGR 转为 HSV
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

    # opencv中，H（色度）范围[0,255],S(饱和度)范围[0,255]，V(亮度)范围[0,115]
    lower_blue = np.array([0, 0, 0])
    upper_blue = np.array([255, 255, 115])

    # 获得区域的mask
    mask = cv2.inRange(hsv, lower_blue, upper_blue)

    h_structure = cv2.getStructuringElement(cv2.MORPH_RECT, (3, 1))
    # 膨胀mask
    mask = cv2.dilate(mask, h_structure, 1)
    cv2.imwrite("mask.jpg", mask)

    dst = cv2.inpaint(frame, mask, 5, cv2.INPAINT_TELEA)

    #cv2.namedWindow('mask', cv2.WINDOW_NORMAL)
    #cv2.imshow('mask', dst)
    cv2.imwrite("binary.png", dst)
    return dst

if __name__=="__main__":
    os.chdir(os.path.dirname(__file__))
    # 由于opencv不支持读取中文路径，用以下方法代替cv2.imread
    frame = cv2.imdecode(np.fromfile('binary3.png', dtype=np.uint8), 1)
    repair(frame)