Merge pull request #249 from luyao777/patch-1

Fix bugs in cuhk02.py
2019-10-29 20:27:22 +09:00 · 2019-10-29 20:27:22 +09:00 · 5fa981b28b
parent 7011bffcb6 db00cf5fc3
commit 5fa981b28b
1 changed files with 2 additions and 2 deletions
--- a/torchreid/data/datasets/image/cuhk02.py
+++ b/torchreid/data/datasets/image/cuhk02.py
@ -80,8 +80,8 @@ class CUHK02(ImageDataset):
                    train.append((impath, pid, camid))
                camid += 1
                
-                # add images to train from cam1
-                for impath in impaths1:
+                # add images to train from cam2
+                for impath in impaths2:
                    pid = osp.basename(impath).split('_')[0]
                    pid = pid2label[pid]
                    train.append((impath, pid, camid))