opencv python照片去背景,从图像opencv python中删除背景色

I have many images of specimen which have uncontrollable background color. Some of them have black background. Some of them have white background. Some of them have green background, etc.

I would like to remove these background color of a given image where the object in the image is just only one specimen. I try this code but it does not work as i expect.

def get_holes(image, thresh):

gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

im_bw = cv2.threshold(gray, thresh, 255, cv2.THRESH_BINARY)[1]

im_bw_inv = cv2.bitwise_not(im_bw)

_, contour, _ = cv2.findContours(im_bw_inv, cv2.RETR_CCOMP, cv2.CHAIN_APPROX_SIMPLE)

for cnt in contour:

cv2.drawContours(im_bw_inv, [cnt], 0, 255, -1)

nt = cv2.bitwise_not(im_bw)

im_bw_inv = cv2.bitwise_or(im_bw_inv, nt)

return im_bw_inv

def remove_background(image, thresh, scale_factor=.25, kernel_range=range(1, 15), border=None):

border = border or kernel_range[-1]

holes = get_holes(image, thresh)

small = cv2.resize(holes, None, fx=scale_factor, fy=scale_factor)

bordered = cv2.copyMakeBorder(small, border, border, border, border, cv2.BORDER_CONSTANT)

for i in kernel_range:

kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2