實現(xiàn)效果

第一張圖為原圖，其余的圖為分割后的圖形

代碼實現(xiàn)：

# -*-coding:utf-8-*-import numpy as npimport cv2#----------------------------------------------------------------------def obj_clip(img, foreground, border): result = [] height ,width = np.shape(img) visited = set() for h in range(height): for w in range(width): if img[h,w] == foreground and not (h,w) in visited:obj = visit(img, height, width, h, w, visited, foreground, border)result.append(obj) return result#----------------------------------------------------------------------def visit(img, height, width, h, w, visited, foreground, border): visited.add((h,w)) result = [(h,w)] if w > 0 and not (h, w-1) in visited: if img[h, w-1] == foreground: result += visit(img, height, width, h, w-1, visited , foreground, border) elif border is not None and img[h, w-1] == border: result.append((h, w-1)) if w < width-1 and not (h, w+1) in visited: if img[h, w+1] == foreground: result += visit(img, height, width, h, w+1, visited, foreground, border) elif border is not None and img[h, w+1] == border: result.append((h, w+1)) if h > 0 and not (h-1, w) in visited: if img[h-1, w] == foreground: result += visit(img, height, width, h-1, w, visited, foreground, border) elif border is not None and img[h-1, w] == border: result.append((h-1, w)) if h < height-1 and not (h+1, w) in visited: if img[h+1, w] == foreground : result += visit(img, height, width, h+1, w, visited, foreground, border) elif border is not None and img[h+1, w] == border: result.append((h+1, w)) return result#----------------------------------------------------------------------if __name__ == '__main__': import cv2 import sys sys.setrecursionlimit(100000) img = np.zeros([400,400]) cv2.rectangle(img, (10,10), (150,150), 1.0, 5) cv2.circle(img, (270,270), 70, 1.0, 5) cv2.line(img, (100,10), (100,150), 0.5, 5) #cv2.putText(img, 'Martin',(200,200), 1.0, 5) cv2.imshow('img', img*255) cv2.waitKey(0) for obj in obj_clip(img, 1.0, 0.5): clip = np.zeros([400, 400]) for h, w in obj: clip[h, w] = 0.2 cv2.imshow('aa', clip*255) cv2.waitKey(0)

總結

到此這篇關于python 使用遞歸的方式實現(xiàn)語義圖片分割的文章就介紹到這了,更多相關python 語義圖片分割內容請搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持好吧啦網(wǎng)！