From 31bb0448810ed93abc6637e395d623bc0da5e54a Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 29 Dec 2020 23:26:11 -0800
Subject: [PATCH] update
---
utils/datasets.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/utils/datasets.py b/utils/datasets.py
index a02fbdc5..37a88631 100755
--- a/utils/datasets.py
+++ b/utils/datasets.py
@@ -466,7 +466,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
# class filtering
assert len(l)
- assert 2 in l[:, 0] or 3 in l[:, 0] or random.random() < 0.06
+ assert 2 in l[:, 0] or 3 in l[:, 0] or (1 in l[:, 0] and random.random() < 0.5) or random.random() < 0.06
x[im_file] = [l, shape]
except Exception as e:
--
GitLab