monai
medical
File size: 476 Bytes
15e1e77
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import json


class CreateImageLabelList:
    def __init__(self, filename):
        self.filename = filename
        fid = open(self.filename, "r")
        self.json_dict = json.load(fid)

    def create_dataset(self, grp):
        image_list = []
        label_list = []
        image_label_list = self.json_dict[grp]
        for _ in image_label_list:
            image_list.append(_["image"])
            label_list.append(_["label"])
        return image_list, label_list