生成马赛克图片并且在对应的标签中添加mask区域，用于二值图训练

最新推荐文章于 2025-03-04 14:59:45 发布

剑峰随心

最新推荐文章于 2025-03-04 14:59:45 发布

阅读量481

点赞数

文章标签： opencv cv powerbuilder interceptor graphviz

本文链接：https://blog.youkuaiyun.com/qq_36190978/article/details/109837307

版权

博客涉及OpenCV、CV、PowerBuilder、Interceptor、Graphviz等信息技术相关内容，但具体内容缺失，推测可能围绕这些技术的应用、特点等方面展开。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

 1 import cv2
 2 import os
 3 import numpy as np
 4 # 只需要调整neighbor 越大马赛克的单位块越大
 5 def do_mosaic(frame, x, y, w, h, neighbor=12):
 6     """
 7     马赛克的实现原理是把图像上某个像素点一定范围邻域内的所有点用邻域内左上像素点的颜色代替，这样可以模糊细节，但是可以保留大体的轮廓。
 8     :param frame: opencv frame
 9     :param int x :  马赛克左顶点
10     :param int y:  马赛克右顶点
11     :param int w:  马赛克宽
12     :param int h:  马赛克高
13     :param int neighbor:  马赛克每一块的宽
14     """
15     fh, fw = frame.shape[0], frame.shape[1]
16     if (y + h > fh) or (x + w > fw):
17         print()
18         return
19     for i in range(0, h - neighbor, neighbor):  # 关键点0 减去neightbour 防止溢出
20         for j in range(0, w - neighbor, neighbor):
21             rect = [j + x, i + y, neighbor, neighbor]
22             color = frame[i + y][j + x].tolist()  # 关键点1 tolist
23             left_up = (rect[0], rect[1])
24             right_down = (rect[0] + neighbor - 1, rect[1] + neighbor - 1)  # 关键点2 减去一个像素
25             cv2.rectangle(frame, left_up, right_down, color, -1)
26     return frame
27 
28 # 原图路径
29 oriRoot = r"D:\jianfeng\ali\generateMask\pic"
30 # mask路径
31 maskRoot = r"D:\jianfeng\ali\generateMask\mask"
32 pics = os.listdir(oriRoot)
33 for pic in pics:
34     # im = cv2.imread("81.jpg", 1)
35     im = cv2.imread(os.path.join(oriRoot, pic), 1)
36 
37     print(os.path.join(oriRoot, pic))
38     print(im)
39     imMask = cv2.imread(os.path.join(maskRoot, pic.split(".")[0]+".png"), 1)
40     hPic = im.shape[0]
41     wPic = im.shape[0]
42     x = np.random.randint(20, hPic-500)
43     y = np.random.randint(20, wPic-500)
44     w = np.random.randint(20, 400)
45     h = np.random.randint(20, 400)
46     imnew = do_mosaic(im, x, y, w, h)
47     print(imnew)
48     imMask[y:y+h, x:x+w] = 255
49 
50     # show
51     cv2.imshow('mosaic', cv2.resize(imnew, (500, 500)))
52     cv2.imshow('mosaicMask', cv2.resize(imMask, (500, 500)))
53     cv2.waitKey(0)