Как выделить объект переднего фона python?

Question

[Pythonchik3]

Всем привет! Мне нужно написать программу, которой на вход подается изображение, а на выходе изображение, на котором убрали задний фон. Желательно это сделать на python. Фактически, надо убрать задний фон изображения. Как это можно сделать?
(пробовал гуглить, но либо не то, либо код из статьи не работает...)

Comments

[U235U235]

Пример картинки? Формализуйте чем отличается объект от фона.

[Влад Григорьев]

если надо выделить человека есть такая модель https://www.npmjs.com/package/@tensorflow-models/b...
там если поискать есть варианты как можно с конвертировать

[Pythonchik3]

U235U235, Например дом близким планом, небо и лес. Дом - передний план. Машина, на фоне стены - машина - передний план. Главный объект фото, обычно расположен по центру.

[Pythonchik3]

U235U235,

import numpy as np
import cv2 as cv
from matplotlib import pyplot as plt

img = cv.imread('img.jpg')
mask = np.zeros(img.shape[:2],np.uint8)

bgdModel = np.zeros((1,65),np.float64)
fgdModel = np.zeros((1,65),np.float64)

rect = (50,50,450,290)

cv.grabCut(img,mask,rect,bgdModel,fgdModel,5,cv.GC_INIT_WITH_RECT)
mask2 = np.where((mask==2)|(mask==0),0,1).astype('uint8')
img = img*mask2[:,:,np.newaxis]
plt.imshow(img),plt.colorbar(),plt.show()

Вот такой код я сделал (по документалке опенцв), но все же он выделяет на всегда идеально.

[U235U235]

Можно попробовать выделять grabCut не центр картинки, а наоборот периметр. С учетом того, что объект обычно не касается края картинки, т.е. есть поля. Т.е. rect будет по очереди верхнем полем, левым, правым, нижним.
Результат будет логическим ИЛИ полученных масок. Надеюсь не сильно сумбурно объяснил идею.

[Pythonchik3]

U235U235, я так и собираюсь делать. Просто сейчас нет под рукой компьютера, а код скопировал с сайта

Answer 1

[Назар Тропанец]

Изпользуй библиотеку opencv, самая лучшая библиотека для таких задач. Как сделать такое не знаю, но могу посоветовать эту библиотеку. Если ничего не получится, можешь написать нейронку для этого))))