mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix typro ##255 

fix #255
pull/299/head
liaoxingyu 2020-09-10 11:10:40 +08:00
parent 74a6938289
commit 1f4e4909eb
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
demo/predictor.py
View File

@ -70,16 +70,16 @@ class FeatureExtractionDemo(object):
if cnt >= buffer_size: if cnt >= buffer_size:
batch = batch_data.popleft() batch = batch_data.popleft()
predictions = self.predictor.get() predictions = self.predictor.get()
yield predictions, batch["targets"].numpy(), batch["camid"].numpy() yield predictions, batch["targets"].numpy(), batch["camids"].numpy()
while len(batch_data): while len(batch_data):
batch = batch_data.popleft() batch = batch_data.popleft()
predictions = self.predictor.get() predictions = self.predictor.get()
yield predictions, batch["targets"].numpy(), batch["camid"].numpy() yield predictions, batch["targets"].numpy(), batch["camids"].numpy()
else: else:
for batch in data_loader: for batch in data_loader:
predictions = self.predictor(batch["images"]) predictions = self.predictor(batch["images"])
yield predictions, batch["targets"].numpy(), batch["camid"].numpy() yield predictions, batch["targets"].numpy(), batch["camids"].numpy()
class AsyncPredictor: class AsyncPredictor: