语义分割　多个COCO2014格式的Json文件合并

最新推荐文章于 2024-08-12 16:28:49 发布

原创

最新推荐文章于 2024-08-12 16:28:49 发布 · 2.4k 阅读

6 ·

CC 4.0 BY-SA版权

本文介绍了如何将多个遵循COCO2014标准的Json注释文件进行有效合并，适用于语义分割任务的数据集整合。

import os
import datetime
import random
import sys
import operator
import math
import numpy as np
import skimage.io
import matplotlib
from matplotlib import pyplot as plt
import cv2
from collections import defaultdict, OrderedDict
import json
#待合并的路径
FILE_DIR = "C:\\Users\\11446\\Desktop\\json\\"

def load_json(filenamejson):
    with open(filenamejson) as f:
        raw_data = json.load(f)
    return raw_data

file_count = 0
files = next(os.walk(FILE_DIR))[2]
for x in range(len(files)):
    #解析文件名字和后缀
    #print(str(files[x]))
    file_suffix = str(files[x]).split(".")[1]
    file_name = str(files[x]).split(".")[0]
#     #过滤类型不对的文件
#     if file_suffix != "json":
#         continue
    #计数
    file_count = file_count + 1
    #组合文件路径
    filenamejson = FILE_DIR + str(files[x])
    #读取文件
    if x == 0:
        #第一个文件作为ro