From d523a603286f189e4570e913f0b39cc001287230 Mon Sep 17 00:00:00 2001 From: Noah <595311942@qq.com> Date: Tue, 2 Nov 2021 16:26:34 +0800 Subject: [PATCH] Delete Checkpoint & update GLN Laternal Layer --- README.md | 4 ++-- lib/modeling/models/gln.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index ebc3ad9..5f00655 100644 --- a/README.md +++ b/README.md @@ -19,8 +19,8 @@ OpenGait is a flexible and extensible gait recognition project provided by the [ | Baseline | 96.3 | 92.2 | 77.6 | [baseline.yaml](config/baseline.yaml) | 64x44 | 12s | 3.78M | | [GaitSet(AAAI2019)](https://arxiv.org/pdf/1811.06186.pdf) | 95.8(95.0) | 90.0(87.2) | 75.4(70.4) | [gaitset.yaml](config/gaitset.yaml) | 64x44 | 11s | 2.59M | | [GaitPart(CVPR2020)](http://home.ustc.edu.cn/~saihui/papers/cvpr2020_gaitpart.pdf) | 96.1(96.2) | 90.7(91.5) | 78.7(78.7) | [gaitpart.yaml](config/gaitpart.yaml) | 64x44 | 22s | 1.20M | -| [GLN*(ECCV2020)](http://home.ustc.edu.cn/~saihui/papers/eccv2020_gln.pdf) | 96.1(95.6) | 92.5(92.0) | 80.4(77.2) | [gln_phase1.yaml](config/gln/gln_phase1.yaml), [gln_phase2.yaml](config/gln/gln_phase2.yaml) | 128x88 | 14s | 9.46M / 15.6214M | -| [GaitGL(ICCV2021)](https://openaccess.thecvf.com/content/ICCV2021/papers/Lin_Gait_Recognition_via_Effective_Global-Local_Feature_Representation_and_Local_Temporal_ICCV_2021_paper.pdf) | 97.5(97.4) | 95.1(94.5) | 83.5(83.6) | [gaitgl.yaml](config/gaitgl.yaml) | 64x44 | 31s | 3.10M | +| [GLN*(ECCV2020)](http://home.ustc.edu.cn/~saihui/papers/eccv2020_gln.pdf) | 96.4(95.6) | 93.1(92.0) | 81.0(77.2) | [gln_phase1.yaml](config/gln/gln_phase1.yaml), [gln_phase2.yaml](config/gln/gln_phase2.yaml) | 128x88 | 14s | 9.46M / 15.6214M | +| [GaitGL(ICCV2021)](https://openaccess.thecvf.com/content/ICCV2021/papers/Lin_Gait_Recognition_via_Effective_Global-Local_Feature_Representation_and_Local_Temporal_ICCV_2021_paper.pdf) | 97.4(97.4) | 94.5(94.5) | 83.8(83.6) | [gaitgl.yaml](config/gaitgl.yaml) | 64x44 | 31s | 3.10M | The results in the parentheses are mentioned in the papers diff --git a/lib/modeling/models/gln.py b/lib/modeling/models/gln.py index 069ac37..d50a262 100644 --- a/lib/modeling/models/gln.py +++ b/lib/modeling/models/gln.py @@ -52,11 +52,11 @@ class GLN(BaseModel): self.sil_stage_2 = SetBlockWrapper(self.sil_stage_2) self.lateral_layer1 = nn.Conv2d( - in_channels[1]*2, lateral_dim, kernel_size=3, stride=1, padding=1, bias=False) + in_channels[1]*2, lateral_dim, kernel_size=1, stride=1, padding=0, bias=False) self.lateral_layer2 = nn.Conv2d( - in_channels[2]*2, lateral_dim, kernel_size=3, stride=1, padding=1, bias=False) + in_channels[2]*2, lateral_dim, kernel_size=1, stride=1, padding=0, bias=False) self.lateral_layer3 = nn.Conv2d( - in_channels[3]*2, lateral_dim, kernel_size=3, stride=1, padding=1, bias=False) + in_channels[3]*2, lateral_dim, kernel_size=1, stride=1, padding=0, bias=False) self.smooth_layer1 = nn.Conv2d( lateral_dim, lateral_dim, kernel_size=3, stride=1, padding=1, bias=False)