mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

bugfix for ClasDataset

pull/456/head
liaoxingyu 2021-04-08 11:01:15 +08:00
parent 55300730e1
commit 0da5917064
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
projects/FastClas/README.md
View File

@ -1,4 +1,4 @@
# FastCls in FastReID
# FastClas in FastReID
This project provides a baseline and example for image classification based on fastreid.

5
projects/FastClas/fastclas/dataset.py
View File

@ -20,8 +20,9 @@ class ClasDataset(Dataset):
for i in img_items:
classes.add(i[1])
self.classes = sorted(list(classes))
self.class_to_idx = {cls_name: i for i, cls_name in enumerate(classes)}
self.classes = list(classes)
self.classes.sort()
self.class_to_idx = {cls_name: i for i, cls_name in enumerate(self.classes)}
self.idx_to_class = {idx: clas for clas, idx in self.class_to_idx.items()}
def __len__(self):