From 25a7f82df7f501abbcbaae921356c442380c0b26 Mon Sep 17 00:00:00 2001 From: liaoxingyu Date: Fri, 5 Jun 2020 11:23:11 +0800 Subject: [PATCH] change style in baseline --- fastreid/modeling/meta_arch/baseline.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/fastreid/modeling/meta_arch/baseline.py b/fastreid/modeling/meta_arch/baseline.py index 78e910d..9613dcc 100644 --- a/fastreid/modeling/meta_arch/baseline.py +++ b/fastreid/modeling/meta_arch/baseline.py @@ -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