mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Temporary-fix: Restrict training to PERSON class for anonymization

pull/13277/head
Bilal Alsallakh 2024-07-26 15:11:04 -07:00 committed by GitHub
parent 55951a47f0
commit 8392b638a5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/dataloaders.py
View File

@ -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: