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

change style in baseline

pull/150/head
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):
if not self.training:
pred_feat = self.inference(batched_inputs)
try:
return pred_feat, batched_inputs["targets"], batched_inputs["camid"]
except Exception:
return pred_feat
try: return pred_feat, batched_inputs["targets"], batched_inputs["camid"]
except Exception: return pred_feat
images = self.preprocess_image(batched_inputs)
targets = batched_inputs["targets"].long()
@ -71,6 +69,7 @@ class Baseline(nn.Module):
"""
# images = [x["images"] for x in batched_inputs]
images = batched_inputs["images"]
# images = batched_inputs
images.sub_(self.pixel_mean).div_(self.pixel_std)
return images