fix typro ##255

fix 
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

View File

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