coco语义分割标注json转png

COCO格式的语义分割数据json，转为png格式，如CityScapes采用的。使用本脚本可以将coco格式转为CityScapes格式的语义标注。语义分割标注转为png图片格式，可便于进行语义分割标注结果可视化，及时查看标注效果。

语义分割标注是对图像每个像素的类别进行标注，采用图片形式标注更加直观。

函数名称为cocojson2png，代码如下所示。

# -*- coding: utf-8 -*-
"""
Created on Fri Apr  8 2022
@author: suiyingy
"""
import cv2
import numpy as np
import os
import json
from collections import defaultdictdef cocojson2png(coco_dir, json_path='instances_train2017.json', cls_type = 'train2017', save_dir = 'res/'):save_path = os.path.join(save_dir, cls_type)if not os.path.exists(save_path):os.makedirs(save_path)annotation_file = os.path.join(coco_dir, 'annotations', json_path)with open(annotation_file, 'r', encoding='utf-8') as annf:annotations = json.load(annf)images = [i['id'] for i in annotations['images']]img_anno = defaultdict(list)for anno in annotations['annotations']:for img_id in images:if anno['image_id'] == img_id:img_anno[img_id].append(anno)imgid_file = {}for im in annotations['images']:imgid_file[im['id']] = im['file_name']for img_idx in img_anno:image = cv2.imread(coco_dir+'images/'+cls_type+'/'+imgid_file[img_idx])h, w, _ = image.shapeinstance_png = np.zeros((h, w), dtype=np.uint8)for idx, ann in enumerate(img_anno[img_idx]):im_mask = np.zeros((h, w), dtype=np.uint8)mask = []for an in ann['segmentation']:ct = np.expand_dims(np.array(an), 0).astype(int)contour = np.stack((ct[:, ::2], ct[:, 1::2])).Tmask.append(contour)imm = cv2.drawContours(im_mask, mask, -1, 1, -1)imm = imm * (1000 * anno['category_id'] + idx)instance_png = instance_png + imminstance_png = np.clip(instance_png,0 ,255)instance_png = np.expand_dims(instance_png,axis=2).repeat(3,axis=2).astype(np.uint8)print(instance_png.shape)cv2.imwrite(os.path.join(save_path, imgid_file[img_idx].split('.')[0]+".png"), instance_png)if __name__ == '__main__':coco_dir = "coco/"cocojson2png(coco_dir, json_path='instances_train2017.json')

更多三维、二维感知算法和金融量化分析算法请关注“乐乐感知学堂”微信公众号，并将持续进行更新。

coco语义分割标注json转png相关推荐

语义分割标注工具——labelme
参考下面网址: 感谢up主:https://www.bilibili.com/video/BV1xT4y1j7A9?from=search&seid=2903997967250679674 在 ...
语义分割标注方式和指标
语义分割标注方式和指标前言分割任务数据集标注方式标注工具评价指标定义计算方式混淆矩阵 Global Acc Mean Acc Mean IoU 总结参考前言这是我看大佬讲语义分割 ...
【工具】Semantic Segmentation Editor 语义分割标注工具
[工具]Semantic Segmentation Editor 语义分割标注工具一.简介二.界面 1.工具栏 ① 上方横排工具栏 Ctrl + Z 撤销 Ctrl + Y 重做 Alt 操纵工具 ...
python工具方法 18 labelme语义分割标注数据批量转换为png
本实例代码主要使用labelme.utils.shapes_to_label方法实现批量将将labelme的语义分割标注数据转换为图片,支持将标注保存为实例分割训练图.语义分割训练图.语义分割+原图效 ...
图像语义分割标注软件EISeg
paddle 语义分割标注软件: https://github.com/PaddleCV-SIG/EISeghttps://github.com/PaddleCV-SIG/EISeg
coco 语义分割_YOLACT++：目前最热门的实时实例分割开源库
点击我爱计算机视觉标星,更快获取CVML新技术 YOLACT 是ICCV 2019 接收的实时实例分割论文 YOLACT: Real-time Instance Segmentation 提出的算法, ...
点云语义分割标注工具及生成训练数据
1.软件的安装 (1)下载后随便你解压到那个文件夹下(软件github地址) github链接 (2)在终端中运行: curl https://install.meteor.c ...
labelme——语义分割标注
labelme安装 1.视频教程安装使用labelme网络上有很多视频教程,这里推荐一位大神的安装教程,地址如下:https://www.bilibili.com/video/BV16b411G7k ...
(18)语义分割--paddle--EISeg自动标注软件的使用和自己数据集的测试
1.主要参考 (1)使用过程,建议先看一下下面博主的视频 eiseg简单教学_哔哩哔哩_bilibili (2)软件使用,主要参考! 百度飞浆EISeg高效交互式标注分割软件的使用教程_Leonard ...

coco语义分割标注json转png

coco语义分割标注json转png相关推荐

最新文章

热门文章