shuoshuoshuo
/
labelme2yolo
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix parsing error when `imageData` is empty (#50)

This commit is contained in:
Wang Xin 2024-03-24 14:59:09 +08:00 committed by GitHub
parent 94382c05f6
commit 513dbc5609
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/labelme2yolo/l2y.py
View File

@ -137,14 +137,14 @@ def save_yolo_image(json_data, json_dir, image_dir, target_dir, target_name):
"""Save yolo image to image_dir_path/target_dir""" """Save yolo image to image_dir_path/target_dir"""
img_path = os.path.join(image_dir, target_dir, target_name) img_path = os.path.join(image_dir, target_dir, target_name)
if json_data["imageData"] is None: if json_data["imageData"]:
img = img_b64_to_arr(json_data["imageData"])
PIL.Image.fromarray(img).save(img_path)
else:
image_name = json_data["imagePath"] image_name = json_data["imagePath"]
src_image_name = os.path.join(json_dir, image_name) src_image_name = os.path.join(json_dir, image_name)
src_image = cv2.imread(src_image_name) src_image = cv2.imread(src_image_name)
cv2.imwrite(img_path, src_image) cv2.imwrite(img_path, src_image)
else:
img = img_b64_to_arr(json_data["imageData"])
PIL.Image.fromarray(img).save(img_path)
return img_path return img_path