我有几个人体步态数据集图片如下:
输入图像
我正试图删除背景,这样我只在结果图像中有人,我正试图从这个人提取剪影下面是我试图获得所需结果的代码:
import cv2
import numpy as np
import matplotlib.pyplot as plt
image = cv2.imread("inputImage.jpg")
image = cv2.cvtColor(image,cv2.COLOR_BGR2RGB)
grayImage = cv2.cvtColor(image,cv2.COLOR_RGB2GRAY)
ret, thresh = cv2.threshold(
grayImage,
120,
255,
cv2.THRESH_BINARY_INV)
plt.imshow(thresh, cmap="gray", vmin=0, vmax=255),plt.show()
"""
nlabels, labels, stats, centroids = cv2.connectedComponentsWithStats(
thresh,
None,
None,
None,
8,
cv2.CV_32S)
sizes = stats[1:,