Fix image orientation in instance_segmentation/labelme2coco.py

examples/instance_segmentation/labelme2coco.py

@@ -82,20 +82,17 @@ def main():
 
  out_ann_file = osp.join(args.output_dir, 'annotations.json')
  label_files = glob.glob(osp.join(args.input_dir, '*.json'))
- for image_id, label_file in enumerate(label_files):
- print('Generating dataset from:', label_file)
- with open(label_file) as f:
- label_data = json.load(f)
+ for image_id, filename in enumerate(label_files):
+ print('Generating dataset from:', filename)
 
- base = osp.splitext(osp.basename(label_file))[0]
+ label_file = labelme.LabelFile(filename=filename)
+
+ base = osp.splitext(osp.basename(filename))[0]
  out_img_file = osp.join(
  args.output_dir, 'JPEGImages', base + '.jpg'
  )
 
- img_file = osp.join(
- osp.dirname(label_file), label_data['imagePath']
- )
- img = np.asarray(PIL.Image.open(img_file).convert('RGB'))
+ img = labelme.utils.img_data_to_arr(label_file.imageData)
  PIL.Image.fromarray(img).save(out_img_file)
  data['images'].append(dict(
  license=0,
@@ -109,7 +106,7 @@ def main():
 
  masks = {} # for area
  segmentations = collections.defaultdict(list) # for segmentation
- for shape in label_data['shapes']:
+ for shape in label_file.shapes:
  points = shape['points']
  label = shape['label']
  group_id = shape.get('group_id')