rm blank space
parent
240a831ba9
commit
d1dbde7cb3
|
@ -96,10 +96,4 @@ class VideoDataset(Dataset):
|
|||
imgs.append(img)
|
||||
imgs = torch.cat(imgs, dim=0)
|
||||
|
||||
return imgs, pid, camid
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
return imgs, pid, camid
|
|
@ -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))
|
|
@ -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))
|
|
@ -262,11 +262,4 @@ def test(model, queryloader, galleryloader, use_gpu, ranks=[1, 5, 10, 20]):
|
|||
return cmc[0]
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
main()
|
|
@ -248,11 +248,4 @@ def test(model, queryloader, galleryloader, use_gpu, ranks=[1, 5, 10, 20]):
|
|||
return cmc[0]
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
main()
|
|
@ -264,11 +264,4 @@ def test(model, queryloader, galleryloader, use_gpu, ranks=[1, 5, 10, 20]):
|
|||
return cmc[0]
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
main()
|
|
@ -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()
|
|
@ -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()
|
|
@ -40,4 +40,4 @@ class Random2DTranslation(object):
|
|||
return croped_img
|
||||
|
||||
if __name__ == '__main__':
|
||||
pass
|
||||
pass
|
Loading…
Reference in New Issue