Водораздел с пользовательскими маркерами в полутоновом изображении
У меня есть серое изображение, которое представляет карту глубины, и это выглядит так: Карта глубины от Kinect Sensor
Я хочу убрать пол, и поэтому я решил сделать это с помощью алгоритма Watershed из OpenCV. Проблема в том, что я не знаю, как определить свой собственный маркер (я буду помещать его внизу изображения где-то посередине), который требуется Watershed. Я пробовал что-то вроде этого, но это не похоже на работу:
import numpy as np
import cv2
from matplotlib import pyplot as plt
img = cv2.imread('depth_floor.png',0)
mask_seed = np.zeros((252, 336), dtype = np.uint8)
mask_seed[251][336/2] = 255
marker32 = np.int32(mask_seed)
res = cv2.watershed(img, marker32)
cv2.imshow("Image without floor", res)
cv2.waitKey(0)
Любая помощь приветствуется!