jison

Fly*Boy

已于 2022-04-08 16:21:48 修改

阅读量369

点赞数

分类专栏：学习文章标签： python

于 2020-11-30 10:56:08 首次发布

本文链接：https://blog.youkuaiyun.com/mvpboy123/article/details/110379724

版权

学习专栏收录该内容

32 篇文章

订阅专栏

该博客展示了如何将肝脏分割任务的dataset.json文件进行重新组织，包括定义数据集名称、描述、标签和文件路径。通过Python脚本，作者逐项填充了文件内容，包括训练和测试图片列表。

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

import json
import os
path ='/mnt/tiger/Zhengsc/vessel/nnUNet_raw/Task01/dataset.json'
name = os.listdir('/mnt/tiger/Zhengsc/vessel/nnUNet_raw/Task01/imagesTr')
name_test = os.listdir('/mnt/tiger/Zhengsc/vessel/nnUNet_raw/Task01/imagesTs')
with open(path, "w+", encoding="utf-8") as  f:
    f.write('{\n')
    f.write('"name": "Liver",\n')
    f.write('"description": "Liver,and cancer segmentation",\n')
    f.write('"reference": "xxxx",\n')
    f.write('"licence": "CC-BY-SA 4.0",\n')
    f.write('"release": "1.0 04/05/2018",\n')
    f.write('"tensorImageSize": "3D",\n')
    f.write('"modality": {\n')
    f.write('"0": "CT"\n')
    f.write('},\n\n')
    f.write('"labels": {\n')
    f.write('"0": "background",\n')
    f.write('"1": "liver"\n},\n')
    f.write('"numTraining": 0,\n')
    f.write('"numTest":1,\n')
    f.write('"training":[')
    beforeI = '{"image":"./imagesTr/'
    beforeL = '","label":"./labelsTr/'
    for i in range(0,302):
        f.write(beforeI)
        f.write(name[i][-24:])
        f.write(beforeL)
        f.write(name[i][-24:])
        f.write('"}')
        print(name[i][-24:])
        if i<302:
            f.write(',')
    f.write('],\n')
    f.write('"test":[')
    for i in range(0,140):
        f.write('"./imagesTs/'+name_test[i][-24:]+'"')
        if i<140:
            f.write(',')
    f.write(']\n}')
print('hello')

#