From 7b3a07bca931b5359cd05639e039e05a0d539c79 Mon Sep 17 00:00:00 2001 From: Chao Fan <92680886+ChaoFan996@users.noreply.github.com> Date: Tue, 19 Mar 2024 15:00:08 +0800 Subject: [PATCH] Update deepgaitv2.py --- opengait/modeling/models/deepgaitv2.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/opengait/modeling/models/deepgaitv2.py b/opengait/modeling/models/deepgaitv2.py index 60894e0..2f23e53 100644 --- a/opengait/modeling/models/deepgaitv2.py +++ b/opengait/modeling/models/deepgaitv2.py @@ -61,7 +61,6 @@ class DeepGaitV2(BaseModel): self.layer3 = SetBlockWrapper(self.layer3) self.layer4 = SetBlockWrapper(self.layer4) - self.mode = mode self.FCs = SeparateFCs(16, channels[3], channels[2]) self.BNNecks = SeparateBNNecks(16, channels[2], class_num=model_cfg['SeparateBNNecks']['class_num']) @@ -93,10 +92,6 @@ class DeepGaitV2(BaseModel): def forward(self, inputs): ipts, labs, typs, vies, seqL = inputs - - if not self.training and len(labs) != 1 and self.mode != '2d': - raise ValueError( - 'The input size of each GPU must be 1 in testing mode, but got {}!'.format(len(labs))) if len(ipts[0].size()) == 4: sils = ipts[0].unsqueeze(1)