图像语义分割出的json文件和原图，用plt绘制图像mask

# -*- coding: utf-8 -*-import osdef cut(filename, start, end): ???assert os.path.exists(filename) is True, "The soruse file is not exists." ???start1 = start.replace(":","") ???end1 = end.replace(":","") ???#print(start1 + " " + end1) ???videoname = "{}{}-{}.mp4".format(filename.rsplit(".",1)[0],start1,end1) ???cmd = "ffmpeg -i {} -vcodec copy -acodec copy -ss {} -to {} {} -y".format(filename,start,end,videoname) ???result = os.popen(cmd) ???return resultif __name__ == "__main__": ???file = input("需要截取的视频:") ???start = input("起始时间(HH:MM:SS):") ???end = input("结束时间(HH:MM:SS):") ???print(cut(file, start, end))

import cv2def get_video_pic(name,zhen): ???path = name.rsplit(".",1)[0] ???cap = cv2.VideoCapture(name) ???for i in range(1,int(cap.get(7)),zhen): ???????cap.set(1, i) ???????rval, frame = cap.read() ???????if rval: ???????????picname = "{}{}.jpg".format(path,str(i)) ???????????cv2.imwrite(picname, frame) ???????cap.release()if __name__ == "__main__": ???video = r"C:/Users/yuanpeng.xie/Desktop/test/yongcun-3.30-3.36.mp4" ???frame = 40 ???get_video_pic(video,int(frame)) ???print("over")

import numpy as npimport matplotlib.pyplot as pltimport matplotlib.image as imgfrom PIL import Imageimport osimport jsondef PictureToMask(d_object, sourcePicture): ???‘‘‘得到原图的宽度和高度‘‘‘ ???im = Image.open(sourcePicture) ???size = list(im.size) ???width = size[0] ???height = size[1] ???????‘‘‘将图片的像素的宽度和高度换算成英寸的宽度和高度‘‘‘ ???dpi = 80 ????????????????#分辨率 ???ycwidth = width/dpi ?????#宽度(英寸) = 像素宽度 / 分辨率 ???ycheight = height/dpi ???#高度(英寸) = 像素高度 / 分辨率 ???color = ["g","r","b","y","skyblue","k","m","c"] ???fig, ax = plt.subplots(figsize=(ycwidth,ycheight)) ???for region in d_object: ???????‘‘‘将传进来的x轴坐标点和y轴坐标点转换成numpy数组，相加后转置成多行两列‘‘‘ ???????x = np.array(d_object[region][0]) ???????y = np.array(d_object[region][1]) * -1 ???????xy = np.vstack([x,y]).T ???????‘‘‘ ???????#设置画框的背景图片为原图 ???????fig = plt.figure(figsize=(ycwidth,ycheight),dpi=dpi) ???????bgimg = img.imread(sourcePicture) ???????fig.figimage(bgimg) ???????‘‘‘ ???????‘‘‘将numpy中的坐标连城线，绘制在plt上‘‘‘ ???????plt.plot(xy[:,0],xy[:,1],color=color[int(region)]) ???????plt.fill_between(xy[:,0],xy[:,1],facecolor=color[int(region)]) ?#对该分割区域填充颜色 ???plt.xticks([0,width]) ???plt.yticks([0,-height]) ???plt.axis("off") ???#保存图片 ???path = sourcePicture.rsplit(".",1)[0] ???print(sourcePicture) ???print(path) ???plt.savefig(path + "-mask.png", format=‘png‘, bbox_inches=‘tight‘, transparent=True, dpi=100) # bbox_inches=‘tight‘ 图片边界空白紧致, 背景透明 ???#plt.show()def getJson(filepath): ???‘‘‘从文件夹获取json文件内容，返回字典‘‘‘ ???files = os.listdir(filepath) ???for file in files: ???????if file.split(".")[1] == "json": ???????????jsonfile = filepath + file ???????????break ???jsonstr = open(jsonfile,"r",encoding="utf8").read() ???d_json = json.loads(jsonstr) ???#print(d_json) ???return d_jsondef getPath(): ???‘‘‘输入图片文件夹路径‘‘‘ ???filepath = input("图片文件夹路径：") ???if filepath.endswith != "/" or filepath.endswith != "\\": ???????filepath = filepath + "/" ???return filepathdef main(): ???filepath = getPath() ???d_json = getJson(filepath) ???for key in d_json: ???????data = d_json.get(key) ???????pictureName = data["filename"] ???????d_object = {} ???????for region in data["regions"]: ???????????l_object = [] ???????????x = data["regions"][region]["shape_attributes"]["all_points_x"] ???????????y = data["regions"][region]["shape_attributes"]["all_points_y"] ???????????l_object.append(x) ???????????l_object.append(y) ???????????d_object[region] = l_object ???????sourcePicture = filepath + pictureName ???????PictureToMask(d_object, sourcePicture)if __name__ == "__main__": ???main()

plt.axes([0,0,1,1])