mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add volatile=True

pull/17/head
KaiyangZhou 2018-03-13 10:23:44 +00:00
parent c8616103e5
commit e0d80aac84
4 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train_img_model_xent.py
View File

@ -193,7 +193,7 @@ def test(model, queryloader, galleryloader, use_gpu, ranks=[1, 5, 10, 20]):
for batch_idx, (imgs, pids, camids) in enumerate(queryloader):
if use_gpu:
imgs = imgs.cuda()
imgs = Variable(imgs)
imgs = Variable(imgs, volatile=True)
features = model(imgs)
features = features.data.cpu()
qf.append(features)
@ -209,7 +209,7 @@ def test(model, queryloader, galleryloader, use_gpu, ranks=[1, 5, 10, 20]):
for batch_idx, (imgs, pids, camids) in enumerate(galleryloader):
if use_gpu:
imgs = imgs.cuda()
imgs = Variable(imgs)
imgs = Variable(imgs, volatile=True)
features = model(imgs)
features = features.data.cpu()
gf.append(features)

4
train_img_model_xent_htri.py
View File

@ -202,7 +202,7 @@ def test(model, queryloader, galleryloader, use_gpu, ranks=[1, 5, 10, 20]):
for batch_idx, (imgs, pids, camids) in enumerate(queryloader):
if use_gpu:
imgs = imgs.cuda()
imgs = Variable(imgs)
imgs = Variable(imgs, volatile=True)
features = model(imgs)
features = features.data.cpu()
qf.append(features)
@ -218,7 +218,7 @@ def test(model, queryloader, galleryloader, use_gpu, ranks=[1, 5, 10, 20]):
for batch_idx, (imgs, pids, camids) in enumerate(galleryloader):
if use_gpu:
imgs = imgs.cuda()
imgs = Variable(imgs)
imgs = Variable(imgs, volatile=True)
features = model(imgs)
features = features.data.cpu()
gf.append(features)

4
train_vid_model_xent.py
View File

@ -201,7 +201,7 @@ def test(model, queryloader, galleryloader, pool, use_gpu, ranks=[1, 5, 10, 20])
for batch_idx, (imgs, pids, camids) in enumerate(queryloader):
if use_gpu:
imgs = imgs.cuda()
imgs = Variable(imgs)
imgs = Variable(imgs, volatile=True)
b, s, c, h, w = imgs.size()
imgs = imgs.view(b*s, c, h, w)
features = model(imgs)
@ -224,7 +224,7 @@ def test(model, queryloader, galleryloader, pool, use_gpu, ranks=[1, 5, 10, 20])
for batch_idx, (imgs, pids, camids) in enumerate(galleryloader):
if use_gpu:
imgs = imgs.cuda()
imgs = Variable(imgs)
imgs = Variable(imgs, volatile=True)
imgs = imgs.view(b*s, c, h, w)
features = model(imgs)
features = features.view(b, s, -1)

4
train_vid_model_xent_htri.py
View File

@ -210,7 +210,7 @@ def test(model, queryloader, galleryloader, pool, use_gpu, ranks=[1, 5, 10, 20])
for batch_idx, (imgs, pids, camids) in enumerate(queryloader):
if use_gpu:
imgs = imgs.cuda()
imgs = Variable(imgs)
imgs = Variable(imgs, volatile=True)
b, s, c, h, w = imgs.size()
imgs = imgs.view(b*s, c, h, w)
features = model(imgs)
@ -233,7 +233,7 @@ def test(model, queryloader, galleryloader, pool, use_gpu, ranks=[1, 5, 10, 20])
for batch_idx, (imgs, pids, camids) in enumerate(galleryloader):
if use_gpu:
imgs = imgs.cuda()
imgs = Variable(imgs)
imgs = Variable(imgs, volatile=True)
imgs = imgs.view(b*s, c, h, w)
features = model(imgs)
features = features.view(b, s, -1)