From 8392b638a5f82ceed15bf168c780a37a0a2a14d4 Mon Sep 17 00:00:00 2001 From: Bilal Alsallakh Date: Fri, 26 Jul 2024 15:11:04 -0700 Subject: [PATCH] Temporary-fix: Restrict training to PERSON class for anonymization --- utils/dataloaders.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: