扫二维码与项目经理沟通
我们在微信上24小时期待你的声音
解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流
创新互联www.cdcxhl.cn八线动态BGP香港云服务器提供商,新人活动买多久送多久,划算不套路!
成都创新互联公司是一家业务范围包括IDC托管业务,虚拟空间、主机租用、主机托管,四川、重庆、广东电信服务器租用,成都天府联通服务器托管,成都网通服务器托管,成都服务器租用,业务范围遍及中国大陆、港澳台以及欧美等多个国家及地区的互联网数据服务公司。这篇文章主要为大家展示了python如何使用递归的方式实现语义图片分割,内容简而易懂,下面让小编带大家一起学习一下吧。
实现效果
第一张图为原图,其余的图为分割后的图形
代码实现:
# -*-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)
我们在微信上24小时期待你的声音
解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流