創(chuàng)新互聯(lián)www.cdcxhl.cn八線動態(tài)BGP香港云服務(wù)器提供商,新人活動買多久送多久,劃算不套路!

這篇文章主要為大家展示了python如何使用遞歸的方式實現(xiàn)語義圖片分割,內(nèi)容簡而易懂,下面讓小編帶大家一起學(xué)習(xí)一下吧。
實現(xiàn)效果

第一張圖為原圖,其余的圖為分割后的圖形
代碼實現(xiàn):
# -*-coding:utf-8-*-
import numpy as np
import 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)語義圖片分割-創(chuàng)新互聯(lián)
轉(zhuǎn)載來源:http://chinadenli.net/article34/dpiepe.html
成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián),為您提供網(wǎng)站維護、定制開發(fā)、網(wǎng)站制作、ChatGPT、網(wǎng)站建設(shè)、標簽優(yōu)化
聲明:本網(wǎng)站發(fā)布的內(nèi)容(圖片、視頻和文字)以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主,如果涉及侵權(quán)請盡快告知,我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場,如需處理請聯(lián)系客服。電話:028-86922220;郵箱:631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載,或轉(zhuǎn)載時需注明來源: 創(chuàng)新互聯(lián)
猜你還喜歡下面的內(nèi)容