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

fix convert caffe model problem

Browse Source
This commit is contained in:
liaoxingyu 2020-06-04 16:39:12 +08:00
parent 12ee19d6ed
commit 94d85fe11c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -48,7 +48,7 @@ class Baseline(nn.Module):
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 KeyError: except Exception:
return pred_feat return pred_feat
images = self.preprocess_image(batched_inputs) images = self.preprocess_image(batched_inputs)