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

bug fixed

This commit is contained in:
greatx 2022-12-27 09:26:27 +08:00
parent 8a5b4d2791
commit 00185f5082
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/labelme2yolo/__about__.py
View File

@ -2,4 +2,4 @@
# #
# SPDX-License-Identifier: MIT # SPDX-License-Identifier: MIT
__version__ = '0.0.7' __version__ = '0.0.8'

11
src/labelme2yolo/l2y.py
View File

@ -93,6 +93,15 @@ def get_label_id_map(json_dir):
return OrderedDict([(label, label_id) for label_id, label in enumerate(label_set)]) return OrderedDict([(label, label_id) for label_id, label in enumerate(label_set)])
def extend_point_list(point_list):
xmin = min([float(point) for point in point_list[::2]])
xmax = max([float(point) for point in point_list[::2]])
ymin = min([float(point) for point in point_list[1::2]])
ymax = max([float(point) for point in point_list[1::2]])
return np.array([xmin, ymin, xmax, ymin, xmax, ymax, xmin, ymax])
def save_yolo_label(json_name, label_dir_path, target_dir, yolo_obj_list): def save_yolo_label(json_name, label_dir_path, target_dir, yolo_obj_list):
txt_path = os.path.join( txt_path = os.path.join(
label_dir_path, target_dir, json_name.replace(".json", ".txt") label_dir_path, target_dir, json_name.replace(".json", ".txt")
@ -287,6 +296,8 @@ class Labelme2YOLO(object):
points = np.zeros(2 * len(point_list)) points = np.zeros(2 * len(point_list))
points[::2] = [float(point[0]) / img_w for point in point_list] points[::2] = [float(point[0]) / img_w for point in point_list]
points[1::2] = [float(point[1]) / img_h for point in point_list] points[1::2] = [float(point[1]) / img_h for point in point_list]
if len(points) == 4:
points = extend_point_list(points)
label_id = self._label_id_map[shape['label']] label_id = self._label_id_map[shape['label']]
return (label_id, points.tolist()) return (label_id, points.tolist())