diff --git a/utils/dataloaders.py b/utils/dataloaders.py index d849d5150..a5a146915 100644 --- a/utils/dataloaders.py +++ b/utils/dataloaders.py @@ -513,7 +513,7 @@ class LoadImagesAndLabels(Dataset): self.indices = range(n) # Update labels - include_class = [] # filter labels to include only these classes (optional) + include_class = [0] # filter labels to include only these classes (optional) include_class_array = np.array(include_class).reshape(1, -1) for i, (label, segment) in enumerate(zip(self.labels, self.segments)): if include_class: