mirrors/deep-person-reid
1
0
Fork 0
mirror of https://github.com/KaiyangZhou/deep-person-reid.git synced 2025-06-03 14:53:23 +08:00
Code Issues Projects Releases Wiki Activity

rm blank space

Browse Source
This commit is contained in:
KaiyangZhou 2018-04-23 10:21:26 +01:00
parent 240a831ba9
commit d1dbde7cb3
10 changed files with 10 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
dataset_loader.py
View File

@ -96,10 +96,4 @@ class VideoDataset(Dataset):
imgs.append(img) imgs.append(img)
imgs = torch.cat(imgs, dim=0) imgs = torch.cat(imgs, dim=0)
return imgs, pid, camid return imgs, pid, camid

8
models/MuDeep.py
View File

@ -182,10 +182,4 @@ class MuDeep(nn.Module):
elif self.loss == {'cent'}: elif self.loss == {'cent'}:
return y, x return y, x
else: else:
raise KeyError("Unsupported loss: {}".format(self.loss)) raise KeyError("Unsupported loss: {}".format(self.loss))

3
models/ResNet.py
View File

@ -85,5 +85,4 @@ class ResNet50M(nn.Module):
elif self.loss == {'cent'}: elif self.loss == {'cent'}:
return prelogits, combofeat return prelogits, combofeat
else: else:
raise KeyError("Unsupported loss: {}".format(self.loss)) raise KeyError("Unsupported loss: {}".format(self.loss))

9
train_img_model_cent.py
View File

@ -262,11 +262,4 @@ def test(model, queryloader, galleryloader, use_gpu, ranks=[1, 5, 10, 20]):
return cmc[0] return cmc[0]
if __name__ == '__main__': if __name__ == '__main__':
main() main()

9
train_img_model_xent.py
View File

@ -248,11 +248,4 @@ def test(model, queryloader, galleryloader, use_gpu, ranks=[1, 5, 10, 20]):
return cmc[0] return cmc[0]
if __name__ == '__main__': if __name__ == '__main__':
main() main()

9
train_img_model_xent_htri.py
View File

@ -264,11 +264,4 @@ def test(model, queryloader, galleryloader, use_gpu, ranks=[1, 5, 10, 20]):
return cmc[0] return cmc[0]
if __name__ == '__main__': if __name__ == '__main__':
main() main()

9
train_vid_model_xent.py
View File

@ -266,11 +266,4 @@ def test(model, queryloader, galleryloader, pool, use_gpu, ranks=[1, 5, 10, 20])
return cmc[0] return cmc[0]
if __name__ == '__main__': if __name__ == '__main__':
main() main()

9
train_vid_model_xent_htri.py
View File

@ -282,11 +282,4 @@ def test(model, queryloader, galleryloader, pool, use_gpu, ranks=[1, 5, 10, 20])
return cmc[0] return cmc[0]
if __name__ == '__main__': if __name__ == '__main__':
main() main()

2
transforms.py
View File

@ -40,4 +40,4 @@ class Random2DTranslation(object):
return croped_img return croped_img
if __name__ == '__main__': if __name__ == '__main__':
pass pass

10
utils.py
View File

@ -87,12 +87,4 @@ def read_json(fpath):
def write_json(obj, fpath): def write_json(obj, fpath):
mkdir_if_missing(osp.dirname(fpath)) mkdir_if_missing(osp.dirname(fpath))
with open(fpath, 'w') as f: with open(fpath, 'w') as f:
json.dump(obj, f, indent=4, separators=(',', ': ')) json.dump(obj, f, indent=4, separators=(',', ': '))