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

bug fixed

This commit is contained in:
Wang Xin 2022-07-26 13:08:34 +08:00
parent 74f7cacf39
commit 9bb17ad4ae
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
labelme2yolo.py
View File

@ -78,6 +78,8 @@ class Labelme2YOLO(object):
train_idxs, val_idxs = train_test_split(range(len(json_names)), train_idxs, val_idxs = train_test_split(range(len(json_names)),
test_size=val_size) test_size=val_size)
tmp_train_len = len(train_idxs) tmp_train_len = len(train_idxs)
test_idxs = []
if test_size > 1e-8:
train_idxs, test_idxs = train_test_split(range(tmp_train_len), test_size=test_size / (1 - val_size)) train_idxs, test_idxs = train_test_split(range(tmp_train_len), test_size=test_size / (1 - val_size))
train_json_names = [json_names[train_idx] for train_idx in train_idxs] train_json_names = [json_names[train_idx] for train_idx in train_idxs]
val_json_names = [json_names[val_idx] for val_idx in val_idxs] val_json_names = [json_names[val_idx] for val_idx in val_idxs]