python 使用遞歸的方式實(shí)現(xiàn)語義圖片分割功能
實(shí)現(xiàn)效果
第一張圖為原圖,其余的圖為分割后的圖形
代碼實(shí)現(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)
總結(jié)
到此這篇關(guān)于python 使用遞歸的方式實(shí)現(xiàn)語義圖片分割的文章就介紹到這了,更多相關(guān)python 語義圖片分割內(nèi)容請(qǐng)搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持好吧啦網(wǎng)!
相關(guān)文章:
1. CSS3實(shí)現(xiàn)動(dòng)態(tài)翻牌效果 仿百度貼吧3D翻牌一次動(dòng)畫特效2. ASP.NET MVC通過勾選checkbox更改select的內(nèi)容3. ASP.NET MVC使用Quartz.NET執(zhí)行定時(shí)任務(wù)4. XMLDOM對(duì)象方法:對(duì)象屬性5. Python在字符串中處理html和xml的方法6. python 錯(cuò)誤處理 assert詳解7. Python使用sftp實(shí)現(xiàn)上傳和下載功能8. Angular獲取ngIf渲染的Dom元素示例9. .NET Core 分布式任務(wù)調(diào)度ScheduleMaster詳解10. Python:chrome.exe的通用webbrowser.get()。open()無法正常工作
