From 0970896aaaa03dd9a1c8c877448877a457707e2a Mon Sep 17 00:00:00 2001 From: Zzier Date: Mon, 8 Jul 2024 20:56:18 +0800 Subject: [PATCH 1/2] Update Scoliosis1K dataset --- README.md | 2 +- opengait/evaluation/evaluator.py | 2 +- opengait/modeling/models/sconet.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 8bd1b12..d40bf4e 100644 --- a/README.md +++ b/README.md @@ -34,7 +34,7 @@ Our team's latest checkpoints for projects such as DeepGaitv2, SkeletonGait, Ske - [Mar 2022] Dataset [GREW](https://www.grew-benchmark.org) is supported in [datasets/GREW](./datasets/GREW). --> ## Our Publications -- [**MICCAI'24**] Gait Patterns as Biomarkers: A Video-Based Approach for Classifying Scoliosis, [*Paper*](https://zhouzi180.github.io/Scoliosis1K)(Coming soon), [*Dataset*](https://zhouzi180.github.io/Scoliosis1K)(Coming soon), and [*Code*](opengait/modeling/models/sconet.py). +- [**MICCAI'24**] Gait Patterns as Biomarkers: A Video-Based Approach for Classifying Scoliosis, [*Paper*](https://zhouzi180.github.io/Scoliosis1K)(Coming soon), [*Dataset*](https://zhouzi180.github.io/Scoliosis1K), and [*Code*](opengait/modeling/models/sconet.py). - [**CVPR'24**] BigGait: Learning Gait Representation You Want by Large Vision Models. [*Paper*](https://arxiv.org/pdf/2402.19122.pdf), and [*Code*](opengait/modeling/models/BigGait.py). - [**AAAI'24**] SkeletonGait++: Gait Recognition Using Skeleton Maps. [*Paper*](https://arxiv.org/pdf/2311.13444.pdf), and [*Code*](opengait/modeling/models/skeletongait%2B%2B.py). - [**AAAI'24**] Cross-Covariate Gait Recognition: A Benchmark. [*Paper*](https://arxiv.org/pdf/2312.14404.pdf), [*Dataset*](https://github.com/ShinanZou/CCGR), and [*Code*](https://github.com/ShiqiYu/OpenGait/blob/master/opengait/modeling/models/deepgaitv2.py). diff --git a/opengait/evaluation/evaluator.py b/opengait/evaluation/evaluator.py index d156898..3f4f6f2 100644 --- a/opengait/evaluation/evaluator.py +++ b/opengait/evaluation/evaluator.py @@ -425,7 +425,7 @@ def evaluate_scoliosis(data, dataset, metric='euc'): class_id = np.array(class_id) # Update class_id with integer labels based on status - class_id_int = np.array([1 if status == 'positive' else 2 if status == 'critical' else 0 for status in class_id]) + class_id_int = np.array([1 if status == 'positive' else 2 if status == 'neutral' else 0 for status in class_id]) print('class_id=', class_id_int) features = np.array(feature) diff --git a/opengait/modeling/models/sconet.py b/opengait/modeling/models/sconet.py index 30efeb9..a977e81 100644 --- a/opengait/modeling/models/sconet.py +++ b/opengait/modeling/models/sconet.py @@ -18,7 +18,7 @@ class ScoNet(BaseModel): def forward(self, inputs): ipts, labs, class_id, _, seqL = inputs - class_id_int = np.array([1 if status == 'positive' else 2 if status == 'critical' else 0 for status in class_id]) + class_id_int = np.array([1 if status == 'positive' else 2 if status == 'neutral' else 0 for status in class_id]) class_id = torch.tensor(class_id_int).cuda() sils = ipts[0] From f806b2bf843410903fdc70556f692e0732ac16e1 Mon Sep 17 00:00:00 2001 From: Zzier Date: Tue, 9 Jul 2024 16:10:49 +0800 Subject: [PATCH 2/2] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index d40bf4e..49300b6 100644 --- a/README.md +++ b/README.md @@ -34,7 +34,7 @@ Our team's latest checkpoints for projects such as DeepGaitv2, SkeletonGait, Ske - [Mar 2022] Dataset [GREW](https://www.grew-benchmark.org) is supported in [datasets/GREW](./datasets/GREW). --> ## Our Publications -- [**MICCAI'24**] Gait Patterns as Biomarkers: A Video-Based Approach for Classifying Scoliosis, [*Paper*](https://zhouzi180.github.io/Scoliosis1K)(Coming soon), [*Dataset*](https://zhouzi180.github.io/Scoliosis1K), and [*Code*](opengait/modeling/models/sconet.py). +- [**MICCAI'24**] Gait Patterns as Biomarkers: A Video-Based Approach for Classifying Scoliosis, [*Paper*](https://arxiv.org/pdf/2407.05726), [*Dataset*](https://zhouzi180.github.io/Scoliosis1K), and [*Code*](opengait/modeling/models/sconet.py). - [**CVPR'24**] BigGait: Learning Gait Representation You Want by Large Vision Models. [*Paper*](https://arxiv.org/pdf/2402.19122.pdf), and [*Code*](opengait/modeling/models/BigGait.py). - [**AAAI'24**] SkeletonGait++: Gait Recognition Using Skeleton Maps. [*Paper*](https://arxiv.org/pdf/2311.13444.pdf), and [*Code*](opengait/modeling/models/skeletongait%2B%2B.py). - [**AAAI'24**] Cross-Covariate Gait Recognition: A Benchmark. [*Paper*](https://arxiv.org/pdf/2312.14404.pdf), [*Dataset*](https://github.com/ShinanZou/CCGR), and [*Code*](https://github.com/ShiqiYu/OpenGait/blob/master/opengait/modeling/models/deepgaitv2.py).