labelme_json_to_dataset不能转换关键点的问题（coordinate list must contain at least 2 coordinates）解决办法

Posted 2021-01-09 missingdays

如果json文件中有关键点，在使用labelme_json_to_dataset.exe时就会有如下错误：

UserWarning: This script is aimed to demonstrate how to convert the JSON file to a single image dataset, and not to handle
multiple JSON files to generate a real-use dataset.
  warnings.warn("This script is aimed to demonstrate how to convert the
"
Traceback (most recent call last):
  File "E:Anaconda3lib
unpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "E:Anaconda3lib
unpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "E:Anaconda3Scriptslabelme_json_to_dataset.exe\__main__.py", line 9, in <module>
  File "E:Anaconda3libsite-packageslabelmeclijson_to_dataset.py", line 62, in main
    lbl = utils.shapes_to_label(img.shape, data[‘shapes‘], label_name_to_value)
  File "E:Anaconda3libsite-packageslabelmeutilsshape.py", line 35, in shapes_to_label
    mask = polygons_to_mask(img_shape[:2], polygons)
  File "E:Anaconda3libsite-packageslabelmeutilsshape.py", line 12, in polygons_to_mask
    PIL.ImageDraw.Draw(mask).polygon(xy=xy, outline=1, fill=1)
  File "E:Anaconda3libsite-packagesPILImageDraw.py", line 185, in polygon
    self.draw.draw_polygon(xy, fill, 1)
TypeError: coordinate list must contain at least 2 coordinates

解决办法是：

打开python目录下的..Libsite-packageslabelmeutilsshape.py

将第8行的polygons_to_mask函数改成如下：

1 def polygons_to_mask(img_shape, polygons):
2     mask = np.zeros(img_shape[:2], dtype=np.uint8)
3     mask = PIL.Image.fromarray(mask)
4     if len(polygons)==1:
5         polygons.append(polygons[0])
6     xy = list(map(tuple, polygons))
7     PIL.ImageDraw.Draw(mask).polygon(xy=xy, outline=1, fill=1)
8     mask = np.array(mask, dtype=bool)
9     return mask