背景
标注完数据集后,想要对图片标注区域增加一些障碍。手动添加障碍非常耗费时间,因此想要通过代码实现对标注区域的自动化设障。
代码实现
1.通过opencv读取图片
img = cv.imread("图片读取路径")
2.生成随机数决定后续添加障碍线的条数,通过for循环再生成两组随机数,分别决定线段的粗细及位置
number2 = random.randint(1, 5)
for x in range(number2):
number1 = random.randint(5, 10)
number = random.randint(0, 300)
cv.line(img, (int(xmin) + number, int(ymin) + number), (int(xmax) - number, int(ymax) - number), (0, 0, 0), number1)
3.保存图片
cv.imwrite("图片保存路径", img)