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

fix pylint ci error

This commit is contained in:
Wang Xin 2023-10-04 16:40:06 +08:00
parent 9f2f443dab
commit 054cfeb3ec
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/labelme2yolo/l2y.py
View File

@ -28,6 +28,7 @@ NUM_THREADS = max(1, os.cpu_count() - 1)
def train_test_split(dataset_index, test_size=0.2):
"""Split dataset into train set and test set with test_size"""
test_size = min(max(0.0, test_size), 1.0)
total_size = len(dataset_index)
train_size = int(math.ceil(total_size * (1.0 - test_size)))
@ -332,7 +333,7 @@ class Labelme2YOLO:
if shape['label'] in self._label_id_map:
label_id = self._label_id_map[shape['label']]
return label_id, yolo_center_x, yolo_center_y, yolo_w, yolo_h
else:
raise f"label {shape['label']} not in {self._label_list}"
def _get_other_shape_yolo_object(self, shape, img_h, img_w):
@ -350,7 +351,7 @@ class Labelme2YOLO:
if shape['label'] in self._label_id_map:
label_id = self._label_id_map[shape['label']]
return label_id, points.tolist()
else:
raise f"label {shape['label']} not in {self._label_list}"
def _save_dataset_yaml(self):