diff --git a/utils/datasets.py b/utils/datasets.py
index a02fbdc5a235edd40f79a5f38f32a860c18b9473..37a886311c75a9770c450414ede5b87341c30a14 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: