From b59e6d3d49cff261213d4bceb60e1324b531e9fc Mon Sep 17 00:00:00 2001 From: greatx Date: Tue, 6 Dec 2022 14:35:15 +0800 Subject: [PATCH] fix bug when test_size not set --- src/labelme2yolo/__about__.py | 2 +- src/labelme2yolo/l2y.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/src/labelme2yolo/__about__.py b/src/labelme2yolo/__about__.py index 7040aae..1761334 100644 --- a/src/labelme2yolo/__about__.py +++ b/src/labelme2yolo/__about__.py @@ -2,4 +2,4 @@ # # SPDX-License-Identifier: MIT -__version__ = '0.0.6' +__version__ = '0.0.7' diff --git a/src/labelme2yolo/l2y.py b/src/labelme2yolo/l2y.py index 55b3aba..3864142 100644 --- a/src/labelme2yolo/l2y.py +++ b/src/labelme2yolo/l2y.py @@ -179,6 +179,8 @@ class Labelme2YOLO(object): test_size=val_size) tmp_train_len = len(train_idxs) test_idxs = [] + if test_size is None: + test_size = 0.0 if test_size > 1e-8: train_idxs, test_idxs = train_test_split( range(tmp_train_len), test_size=test_size / (1 - val_size))