polygon 画图

最新推荐文章于 2024-06-05 13:42:34 发布

转载最新推荐文章于 2024-06-05 13:42:34 发布 · 1k 阅读

文章标签：

#python #json

本文介绍如何利用Cityscape数据集中的json文件绘制道路区域。通过解析json文件中的polygon数据，使用Python API实现道路边界的绘制，并进一步展示如何填充整个道路区域。

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

cityscape数据集，我现在想根据json文件中的polygon画出整个ｒｏａｄ的区域，这是运行的脚本。这个文件必须使用ｃｏｃｏ的pythonAPI的包，把这个脚本放在pythonAPI文件夹下就可以运行。

from pycocotools import mask as maskUtils
import numpy as np
import json
import matplotlib.pyplot as plt

import Image,ImageDraw


with open('aachen_000000_000019_gtFine_polygons.json', 'r') as f:
    data = json.load(f)

# print data['objects'][2]
l = []
for i in data['objects'][0]['polygon']:
    l.extend(i)


img = Image.open("aachen_000000_000019_leftImg8bit.png")
draw = ImageDraw.Draw(img)
length = len(l)
print length
x1 = l[0]
y1 = l[1]
for i in range(1,length/2):
    x2 = l[i*2]
    y2 = l[i*2 + 1]
    draw.line(((x1,y1),(x2,y2)),fill=255)
    x1 = x2
    y1 = y2
img.save("cross_line.jpg")

rles = maskUtils.frPyObjects([l], data['imgHeight'], data['imgWidth'])
rle = maskUtils.merge(rles)
m = maskUtils.decode(rle)
plt.imshow(m)
plt.savefig("filename.png")
plt.show()