Некоторые поврежденные части отсутствуют на изображении (используется обработка изображений с python) - PullRequest
0 голосов
/ 28 февраля 2020

Я пытаюсь отобразить поврежденные части изображения, чтобы восстановить их. Я совершенно новичок в обработке изображений, и я использую cv2. Вот изображения, над которыми я работаю.

оригинальное поврежденное изображение

необходимое изображение маски

И я получить следующее изображение:

Вывод

import cv2
dim = (512,512)
im_color = cv2.imread("couple.png", cv2.IMREAD_COLOR)
im_color = cv2.resize(im_color,dim)
im_gray = cv2.cvtColor(im_color, cv2.COLOR_BGR2GRAY)
mask = cv2.threshold(im_gray, thresh=200, maxval=255, type=cv2.THRESH_BINARY)
im_thresh_gray = cv2.bitwise_and(im_gray, mask[0])
mask3 = cv2.cvtColor(mask[1], cv2.COLOR_GRAY2BGR)
im_thresh_color = cv2.bitwise_and(im_color, mask3)
cv2.imshow("original image", im_color)
cv2.imshow("binary mask", mask[1])
cv2.imshow("3 channel mask", mask3)
cv2.imshow("im_thresh_gray", im_thresh_gray)
cv2.imshow("im_thresh_color", im_thresh_color)
cv2.imwrite("repaired.png",mask3)
cv2.waitKey(0)

Некоторые из поврежденных частей отсутствуют. пожалуйста, кто-нибудь может мне помочь?!

...