mirrors/fast-reid
1
0
Fork 0
mirror of https://github.com/JDAI-CV/fast-reid.git synced 2025-06-03 14:50:47 +08:00
Code Issues Projects Releases Wiki Activity

change style in baseline

Browse Source
This commit is contained in:
liaoxingyu 2020-06-05 11:23:11 +08:00
parent bc221cb05f
commit 25a7f82df7
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
fastreid/modeling/meta_arch/baseline.py
View File

@ -46,10 +46,8 @@ class Baseline(nn.Module):
def forward(self, batched_inputs): def forward(self, batched_inputs):
if not self.training: if not self.training:
pred_feat = self.inference(batched_inputs) pred_feat = self.inference(batched_inputs)
try: try: return pred_feat, batched_inputs["targets"], batched_inputs["camid"]
return pred_feat, batched_inputs["targets"], batched_inputs["camid"] except Exception: return pred_feat
except Exception:
return pred_feat
images = self.preprocess_image(batched_inputs) images = self.preprocess_image(batched_inputs)
targets = batched_inputs["targets"].long() targets = batched_inputs["targets"].long()
@ -71,6 +69,7 @@ class Baseline(nn.Module):
""" """
# images = [x["images"] for x in batched_inputs] # images = [x["images"] for x in batched_inputs]
images = batched_inputs["images"] images = batched_inputs["images"]
# images = batched_inputs
images.sub_(self.pixel_mean).div_(self.pixel_std) images.sub_(self.pixel_mean).div_(self.pixel_std)
return images return images