Skip to content

Commit 9c634e9

Browse files
authored
fix a bug in export_coco_as_yolov5 (#262)
* fix a bug in `export_coco_as_yolov5` * update to v0.8.8
1 parent 91e8b80 commit 9c634e9

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

sahi/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = "0.8.7"
1+
__version__ = "0.8.8"

sahi/utils/coco.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -2230,9 +2230,7 @@ def export_coco_as_yolov5(
22302230
ValueError("'train_coco' have to be provided")
22312231

22322232
# check train_split_rate
2233-
if 0 < train_split_rate and train_split_rate < 1:
2234-
split_mode = "SPLIT"
2235-
else:
2233+
if split_mode and not (0 < train_split_rate < 1):
22362234
ValueError("train_split_rate cannot be <0 or >1")
22372235

22382236
# split dataset

0 commit comments

Comments
 (0)