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

bug fixed (#21)

This commit is contained in:
Wang Xin 2023-05-05 13:01:26 +08:00 committed by GitHub
parent 9370343e5e
commit df4f172722
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -33,6 +33,8 @@ pip install labelme2yolo
**--output\_format (Optional)** The output format of label. **--output\_format (Optional)** The output format of label.
**--label\_list (Optional)** The pre-assigned category labels.
## How to Use ## How to Use
### 1. Convert JSON files, split training, validation and test dataset by --val\_size and --test\_size ### 1. Convert JSON files, split training, validation and test dataset by --val\_size and --test\_size

15
src/labelme2yolo/l2y.py
View File

@ -118,11 +118,20 @@ def save_yolo_label(json_name, label_dir_path, target_dir, yolo_obj_list):
f.write(yolo_obj_line) f.write(yolo_obj_line)
def save_yolo_image(json_data, json_name, image_dir_path, target_dir): def save_yolo_image(json_data, json_path, image_dir_path, target_dir):
json_name = os.path.basename(json_path)
img_name = json_name.replace(".json", ".png") img_name = json_name.replace(".json", ".png")
# make image_path and save image
img_path = os.path.join(image_dir_path, target_dir, img_name) img_path = os.path.join(image_dir_path, target_dir, img_name)
if not os.path.exists(img_path): if json_data["imageData"] is None:
dirname = os.path.dirname(json_path)
image_name = json_data["imagePath"]
src_image_name = os.path.join(dirname, image_name)
src_image = cv2.imread(src_image_name)
cv2.imwrite(img_path, src_image)
else:
img = img_b64_to_arr(json_data["imageData"]) img = img_b64_to_arr(json_data["imageData"])
PIL.Image.fromarray(img).save(img_path) PIL.Image.fromarray(img).save(img_path)
@ -228,7 +237,7 @@ class Labelme2YOLO(object):
(json_name, target_dir.replace('/', ''))) (json_name, target_dir.replace('/', '')))
img_path = save_yolo_image(json_data, img_path = save_yolo_image(json_data,
json_name, json_path,
self._image_dir_path, self._image_dir_path,
target_dir) target_dir)
yolo_obj_list = self._get_yolo_object_list(json_data, img_path) yolo_obj_list = self._get_yolo_object_list(json_data, img_path)