rm blank space

pull/17/head
KaiyangZhou 2018-04-23 10:21:26 +01:00
parent 240a831ba9
commit d1dbde7cb3
10 changed files with 10 additions and 66 deletions

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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