| 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 | |