From 084eba2c578700321ca13789f937123b31f479b9 Mon Sep 17 00:00:00 2001 From: wusize Date: Wed, 29 Jun 2022 16:38:11 +0800 Subject: [PATCH 01/10] 'reformulate_multiview_pose' --- mmpose/models/detectors/multiview_pose.py | 31 +++++++++++++++++------ 1 file changed, 23 insertions(+), 8 deletions(-) diff --git a/mmpose/models/detectors/multiview_pose.py b/mmpose/models/detectors/multiview_pose.py index cfa5e32ea4..d4a26b605a 100644 --- a/mmpose/models/detectors/multiview_pose.py +++ b/mmpose/models/detectors/multiview_pose.py @@ -14,7 +14,7 @@ from mmpose.core.post_processing.post_transforms import ( affine_transform_torch, get_affine_transform) from .. import builder -from ..builder import POSENETS +from ..builder import BACKBONES, HEADS, POSENETS from ..utils.misc import torch_meshgrid_ij from .base import BasePose @@ -150,6 +150,7 @@ class DetectAndRegress(BasePose): def __init__(self, backbone, + keypoint_head, human_detector, pose_regressor, train_cfg=None, @@ -158,11 +159,16 @@ def __init__(self, freeze_2d=True): super(DetectAndRegress, self).__init__() if backbone is not None: - self.backbone = builder.build_posenet(backbone) - if self.training and pretrained is not None: - load_checkpoint(self.backbone, pretrained) + self.backbone = BACKBONES.build(backbone) else: self.backbone = None + if keypoint_head is not None: + self.keypoint_head = HEADS.build(keypoint_head) + else: + self.keypoint_head = None + + if self.training and pretrained is not None: + load_checkpoint(self, pretrained) self.freeze_2d = freeze_2d self.human_detector = builder.MODELS.build(human_detector) @@ -188,8 +194,11 @@ def train(self, mode=True): Module: self """ super().train(mode) - if mode and self.freeze_2d and self.backbone is not None: - self._freeze(self.backbone) + if mode and self.freeze_2d: + if self.backbone is not None: + self._freeze(self.backbone) + if self.keypoint_head is not None: + self._freeze(self.keypoint_head) return self @@ -283,6 +292,12 @@ def train_step(self, data_batch, optimizer, **kwargs): return outputs + def predict_heatmap(self, img): + output = self.backbone(img) + output = self.keypoint_head(output) + + return output + def forward_train(self, img, img_metas, @@ -331,7 +346,7 @@ def forward_train(self, feature_maps = [] assert isinstance(img, list) for img_ in img: - feature_maps.append(self.backbone.forward_dummy(img_)[0]) + feature_maps.append(self.predict_heatmap(img_)[0]) losses = dict() human_candidates, human_loss = self.human_detector.forward_train( @@ -400,7 +415,7 @@ def forward_test( feature_maps = [] assert isinstance(img, list) for img_ in img: - feature_maps.append(self.backbone.forward_dummy(img_)[0]) + feature_maps.append(self.predict_heatmap(img_)[0]) human_candidates = self.human_detector.forward_test( None, img_metas, feature_maps) From 46e60ee9f866a189d32666f13a21bb5dfb7e3b34 Mon Sep 17 00:00:00 2001 From: wusize Date: Thu, 30 Jun 2022 19:09:46 +0800 Subject: [PATCH 02/10] 'rewrite_cfg' --- ...e_prn64x64x64_cpn80x80x20_panoptic_cam5.py | 52 +++++++------------ mmpose/models/detectors/multiview_pose.py | 4 +- 2 files changed, 21 insertions(+), 35 deletions(-) diff --git a/configs/body/3d_kpt_mview_rgb_img/voxelpose/panoptic/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.py b/configs/body/3d_kpt_mview_rgb_img/voxelpose/panoptic/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.py index 90996e1eef..8dd02a8183 100644 --- a/configs/body/3d_kpt_mview_rgb_img/voxelpose/panoptic/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.py +++ b/configs/body/3d_kpt_mview_rgb_img/voxelpose/panoptic/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.py @@ -65,44 +65,30 @@ subset='validation')) # model settings -backbone = dict( - type='AssociativeEmbedding', - pretrained=None, - backbone=dict(type='ResNet', depth=50), - keypoint_head=dict( - type='DeconvHead', - in_channels=2048, - out_channels=num_joints, - num_deconv_layers=3, - num_deconv_filters=(256, 256, 256), - num_deconv_kernels=(4, 4, 4), - loss_keypoint=dict( - type='MultiLossFactory', - num_joints=15, - num_stages=1, - ae_loss_type='exp', - with_ae_loss=[False], - push_loss_factor=[0.001], - pull_loss_factor=[0.001], - with_heatmaps_loss=[True], - heatmaps_loss_factor=[1.0], - )), - train_cfg=dict(), - test_cfg=dict( - num_joints=num_joints, - nms_kernel=None, - nms_padding=None, - tag_per_joint=None, - max_num_people=None, - detection_threshold=None, - tag_threshold=None, - use_detection_val=None, - ignore_too_much=None, +backbone = dict(type='ResNet', depth=50) +keypoint_head = dict( + type='DeconvHead', + in_channels=2048, + out_channels=num_joints, + num_deconv_layers=3, + num_deconv_filters=(256, 256, 256), + num_deconv_kernels=(4, 4, 4), + loss_keypoint=dict( + type='MultiLossFactory', + num_joints=15, + num_stages=1, + ae_loss_type='exp', + with_ae_loss=[False], + push_loss_factor=[0.001], + pull_loss_factor=[0.001], + with_heatmaps_loss=[True], + heatmaps_loss_factor=[1.0], )) model = dict( type='DetectAndRegress', backbone=backbone, + keypoint_head=keypoint_head, pretrained='checkpoints/resnet_50_deconv.pth.tar', human_detector=dict( type='VoxelCenterDetector', diff --git a/mmpose/models/detectors/multiview_pose.py b/mmpose/models/detectors/multiview_pose.py index d4a26b605a..94c9e49510 100644 --- a/mmpose/models/detectors/multiview_pose.py +++ b/mmpose/models/detectors/multiview_pose.py @@ -366,8 +366,8 @@ def forward_train(self, heatmaps_tensor = torch.cat(feature_maps, dim=0) targets_tensor = torch.cat(targets, dim=0) masks_tensor = torch.cat(masks, dim=0) - losses_2d_ = self.backbone.get_loss(heatmaps_tensor, - targets_tensor, masks_tensor) + losses_2d_ = self.keypoint_head.get_loss(heatmaps_tensor, + targets_tensor, masks_tensor) for k, v in losses_2d_.items(): losses_2d[k + '_2d'] = v losses.update(losses_2d) From 37c12f3e98c14892506203b91586035afa08ff76 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=90=B4=E6=80=9D=E6=B3=BD?= Date: Tue, 12 Jul 2022 23:59:33 +0800 Subject: [PATCH 03/10] rewrite_comments --- mmpose/models/detectors/multiview_pose.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/mmpose/models/detectors/multiview_pose.py b/mmpose/models/detectors/multiview_pose.py index 94c9e49510..25e4854e65 100644 --- a/mmpose/models/detectors/multiview_pose.py +++ b/mmpose/models/detectors/multiview_pose.py @@ -138,7 +138,8 @@ class DetectAndRegress(BasePose): """DetectAndRegress approach for multiview human pose detection. Args: - backbone (ConfigDict): Dictionary to construct the 2D pose detector + backbone (ConfigDict): Dictionary to construct the backbone + keypoint_head (ConfigDict): Dictionary to construct the 2d keypoint head human_detector (ConfigDict): dictionary to construct human detector pose_regressor (ConfigDict): dictionary to construct pose regressor train_cfg (ConfigDict): Config for training. Default: None. @@ -521,7 +522,7 @@ def forward_dummy(self, img, input_heatmaps=None, num_candidates=5): feature_maps = [] assert isinstance(img, list) for img_ in img: - feature_maps.append(self.backbone.forward_dummy(img_)[0]) + feature_maps.append(self.predict_heatmap(img_)[0]) _ = self.human_detector.forward_dummy(feature_maps) From 17f29416dea2de56aa0f362ffbf87ce1b7c56889 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=90=B4=E6=80=9D=E6=B3=BD?= Date: Wed, 13 Jul 2022 00:05:10 +0800 Subject: [PATCH 04/10] campus_shelf --- .../campus/voxelpose_prn32x32x32_cpn80x80x20_campus_cam3.py | 1 + .../campus/voxelpose_prn64x64x64_cpn80x80x20_campus_cam3.py | 1 + .../shelf/voxelpose_prn32x32x32_cpn48x48x12_shelf_cam5.py | 1 + .../shelf/voxelpose_prn64x64x64_cpn80x80x20_shelf_cam5.py | 1 + 4 files changed, 4 insertions(+) diff --git a/configs/body/3d_kpt_mview_rgb_img/voxelpose/campus/voxelpose_prn32x32x32_cpn80x80x20_campus_cam3.py b/configs/body/3d_kpt_mview_rgb_img/voxelpose/campus/voxelpose_prn32x32x32_cpn80x80x20_campus_cam3.py index 80d88cbf03..ad95c7b2ae 100644 --- a/configs/body/3d_kpt_mview_rgb_img/voxelpose/campus/voxelpose_prn32x32x32_cpn80x80x20_campus_cam3.py +++ b/configs/body/3d_kpt_mview_rgb_img/voxelpose/campus/voxelpose_prn32x32x32_cpn80x80x20_campus_cam3.py @@ -94,6 +94,7 @@ type='DetectAndRegress', backbone=None, pretrained=None, + keypoint_head=None, human_detector=dict( type='VoxelCenterDetector', image_size=image_size, diff --git a/configs/body/3d_kpt_mview_rgb_img/voxelpose/campus/voxelpose_prn64x64x64_cpn80x80x20_campus_cam3.py b/configs/body/3d_kpt_mview_rgb_img/voxelpose/campus/voxelpose_prn64x64x64_cpn80x80x20_campus_cam3.py index 11df9dc3a5..183e3cbb23 100644 --- a/configs/body/3d_kpt_mview_rgb_img/voxelpose/campus/voxelpose_prn64x64x64_cpn80x80x20_campus_cam3.py +++ b/configs/body/3d_kpt_mview_rgb_img/voxelpose/campus/voxelpose_prn64x64x64_cpn80x80x20_campus_cam3.py @@ -94,6 +94,7 @@ model = dict( type='DetectAndRegress', backbone=None, + keypoint_head=None, pretrained=None, human_detector=dict( type='VoxelCenterDetector', diff --git a/configs/body/3d_kpt_mview_rgb_img/voxelpose/shelf/voxelpose_prn32x32x32_cpn48x48x12_shelf_cam5.py b/configs/body/3d_kpt_mview_rgb_img/voxelpose/shelf/voxelpose_prn32x32x32_cpn48x48x12_shelf_cam5.py index 0eec22a22a..e30cee5903 100644 --- a/configs/body/3d_kpt_mview_rgb_img/voxelpose/shelf/voxelpose_prn32x32x32_cpn48x48x12_shelf_cam5.py +++ b/configs/body/3d_kpt_mview_rgb_img/voxelpose/shelf/voxelpose_prn32x32x32_cpn48x48x12_shelf_cam5.py @@ -91,6 +91,7 @@ model = dict( type='DetectAndRegress', backbone=None, + keypoint_head=None, pretrained=None, human_detector=dict( type='VoxelCenterDetector', diff --git a/configs/body/3d_kpt_mview_rgb_img/voxelpose/shelf/voxelpose_prn64x64x64_cpn80x80x20_shelf_cam5.py b/configs/body/3d_kpt_mview_rgb_img/voxelpose/shelf/voxelpose_prn64x64x64_cpn80x80x20_shelf_cam5.py index 9e4b5c376d..f08a7f171e 100644 --- a/configs/body/3d_kpt_mview_rgb_img/voxelpose/shelf/voxelpose_prn64x64x64_cpn80x80x20_shelf_cam5.py +++ b/configs/body/3d_kpt_mview_rgb_img/voxelpose/shelf/voxelpose_prn64x64x64_cpn80x80x20_shelf_cam5.py @@ -92,6 +92,7 @@ type='DetectAndRegress', backbone=None, pretrained=None, + keypoint_head=None, human_detector=dict( type='VoxelCenterDetector', image_size=image_size, From 5e7bbe743aca48464399d4fa1c5dc80cc6a9152f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=90=B4=E6=80=9D=E6=B3=BD?= Date: Wed, 13 Jul 2022 12:08:59 +0800 Subject: [PATCH 05/10] unit_test --- mmpose/models/detectors/multiview_pose.py | 8 +++++--- tests/test_models/test_multiview_pose.py | 1 + 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/mmpose/models/detectors/multiview_pose.py b/mmpose/models/detectors/multiview_pose.py index 25e4854e65..a16ef7accf 100644 --- a/mmpose/models/detectors/multiview_pose.py +++ b/mmpose/models/detectors/multiview_pose.py @@ -138,8 +138,9 @@ class DetectAndRegress(BasePose): """DetectAndRegress approach for multiview human pose detection. Args: - backbone (ConfigDict): Dictionary to construct the backbone - keypoint_head (ConfigDict): Dictionary to construct the 2d keypoint head + backbone (ConfigDict): Dictionary to construct the backbone. + keypoint_head (ConfigDict): Dictionary to construct the 2d + keypoint head. human_detector (ConfigDict): dictionary to construct human detector pose_regressor (ConfigDict): dictionary to construct pose regressor train_cfg (ConfigDict): Config for training. Default: None. @@ -368,7 +369,8 @@ def forward_train(self, targets_tensor = torch.cat(targets, dim=0) masks_tensor = torch.cat(masks, dim=0) losses_2d_ = self.keypoint_head.get_loss(heatmaps_tensor, - targets_tensor, masks_tensor) + targets_tensor, + masks_tensor) for k, v in losses_2d_.items(): losses_2d[k + '_2d'] = v losses.update(losses_2d) diff --git a/tests/test_models/test_multiview_pose.py b/tests/test_models/test_multiview_pose.py index d37ded8619..880b1dfe36 100644 --- a/tests/test_models/test_multiview_pose.py +++ b/tests/test_models/test_multiview_pose.py @@ -63,6 +63,7 @@ def test_voxelpose_forward(): model_cfg = dict( type='DetectAndRegress', backbone=None, + keypoint_head=None, human_detector=dict( type='VoxelCenterDetector', image_size=[960, 512], From f254dcb8f9d81d1e588a3f224f3ea472a407db3d Mon Sep 17 00:00:00 2001 From: wusize Date: Thu, 30 Jun 2022 19:09:46 +0800 Subject: [PATCH 06/10] 'rewrite_cfg' --- mmpose/models/detectors/multiview_pose.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mmpose/models/detectors/multiview_pose.py b/mmpose/models/detectors/multiview_pose.py index a16ef7accf..9e3710156d 100644 --- a/mmpose/models/detectors/multiview_pose.py +++ b/mmpose/models/detectors/multiview_pose.py @@ -369,8 +369,12 @@ def forward_train(self, targets_tensor = torch.cat(targets, dim=0) masks_tensor = torch.cat(masks, dim=0) losses_2d_ = self.keypoint_head.get_loss(heatmaps_tensor, +<<<<<<< HEAD targets_tensor, masks_tensor) +======= + targets_tensor, masks_tensor) +>>>>>>> 'rewrite_cfg' for k, v in losses_2d_.items(): losses_2d[k + '_2d'] = v losses.update(losses_2d) From cce03a11ddd38a561406904e9d49f9a8b191d9a0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=90=B4=E6=80=9D=E6=B3=BD?= Date: Tue, 12 Jul 2022 23:59:33 +0800 Subject: [PATCH 07/10] rewrite_comments --- mmpose/models/detectors/multiview_pose.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/mmpose/models/detectors/multiview_pose.py b/mmpose/models/detectors/multiview_pose.py index 9e3710156d..a16ef7accf 100644 --- a/mmpose/models/detectors/multiview_pose.py +++ b/mmpose/models/detectors/multiview_pose.py @@ -369,12 +369,8 @@ def forward_train(self, targets_tensor = torch.cat(targets, dim=0) masks_tensor = torch.cat(masks, dim=0) losses_2d_ = self.keypoint_head.get_loss(heatmaps_tensor, -<<<<<<< HEAD targets_tensor, masks_tensor) -======= - targets_tensor, masks_tensor) ->>>>>>> 'rewrite_cfg' for k, v in losses_2d_.items(): losses_2d[k + '_2d'] = v losses.update(losses_2d) From 5e96ae486163d8e479ac953d42559a8ecb1431c4 Mon Sep 17 00:00:00 2001 From: wusize Date: Wed, 18 Jan 2023 21:31:11 +0800 Subject: [PATCH 08/10] readme --- .../voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.md | 2 +- .../voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.yml | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/configs/body/3d_kpt_mview_rgb_img/voxelpose/panoptic/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.md b/configs/body/3d_kpt_mview_rgb_img/voxelpose/panoptic/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.md index b473572d59..4164438469 100644 --- a/configs/body/3d_kpt_mview_rgb_img/voxelpose/panoptic/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.md +++ b/configs/body/3d_kpt_mview_rgb_img/voxelpose/panoptic/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.md @@ -34,4 +34,4 @@ Results on CMU Panoptic dataset. | Arch | mAP | mAR | MPJPE | Recall@500mm | ckpt | log | | :--------------------------------------------------------- | :---: | :---: | :---: | :----------: | :--------------------------------------------------------: | :-------------------------------------------------------: | -| [prn64_cpn80_res50](/configs/body/3d_kpt_mview_rgb_img/voxelpose/panoptic/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.py) | 97.31 | 97.99 | 17.57 | 99.85 | [ckpt](https://download.openmmlab.com/mmpose/body3d/voxelpose/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5-545c150e_20211103.pth) | [log](https://download.openmmlab.com/mmpose/body3d/voxelpose/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5_20211103.log.json) | +| [prn64_cpn80_res50](/configs/body/3d_kpt_mview_rgb_img/voxelpose/panoptic/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.py) | 97.15 | 97.70 | 17.09 | 99.25 | [ckpt](https://download.openmmlab.com/mmpose/body3d/voxelpose/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5-358648cb_20230118.pth) | [log](https://download.openmmlab.com/mmpose/body3d/voxelpose/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5_20230118.log.json) | diff --git a/configs/body/3d_kpt_mview_rgb_img/voxelpose/panoptic/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.yml b/configs/body/3d_kpt_mview_rgb_img/voxelpose/panoptic/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.yml index 8b5e57897f..8a67e35190 100644 --- a/configs/body/3d_kpt_mview_rgb_img/voxelpose/panoptic/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.yml +++ b/configs/body/3d_kpt_mview_rgb_img/voxelpose/panoptic/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5.yml @@ -15,8 +15,8 @@ Models: Results: - Dataset: CMU Panoptic Metrics: - MPJPE: 17.57 - mAP: 97.31 - mAR: 97.99 + MPJPE: 17.09 + mAP: 97.15 + mAR: 97.7 Task: Body 3D Keypoint - Weights: https://download.openmmlab.com/mmpose/body3d/voxelpose/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5-545c150e_20211103.pth + Weights: https://download.openmmlab.com/mmpose/body3d/voxelpose/voxelpose_prn64x64x64_cpn80x80x20_panoptic_cam5-358648cb_20230118.pth From 528d4dc842036f4b71460cc8c4fcbc1619bd56e4 Mon Sep 17 00:00:00 2001 From: wusize Date: Wed, 15 Feb 2023 12:22:10 +0800 Subject: [PATCH 09/10] 'pre_commit' --- model-index.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model-index.yml b/model-index.yml index ed2300dbd2..03c8811120 100644 --- a/model-index.yml +++ b/model-index.yml @@ -109,8 +109,8 @@ Import: - configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/wflw/hrnetv2_dark_wflw.yml - configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/wflw/hrnetv2_wflw.yml - configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/resnet_deepfashion.yml -- configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/resnet_deepfashion.yml - configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion2/resnet_deepfashion2.yml +- configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/resnet_deepfashion.yml - configs/hand/2d_kpt_sview_rgb_img/deeppose/onehand10k/resnet_onehand10k.yml - configs/hand/2d_kpt_sview_rgb_img/deeppose/panoptic2d/resnet_panoptic2d.yml - configs/hand/2d_kpt_sview_rgb_img/deeppose/rhd2d/resnet_rhd2d.yml From 538b2bbde89b9c85ba282872235354fc3bad4a74 Mon Sep 17 00:00:00 2001 From: wusize Date: Wed, 15 Mar 2023 21:33:59 +0800 Subject: [PATCH 10/10] 'replace_mat_with_npy' --- .../base/kpt_3d_mview_rgb_img_direct_dataset.py | 6 +----- tests/data/campus/actorsGT.mat | Bin 4920 -> 0 bytes tests/data/campus/actorsGT.npy | Bin 0 -> 1136 bytes tests/data/shelf/actorsGT.mat | Bin 13968 -> 0 bytes tests/data/shelf/actorsGT.npy | Bin 0 -> 2144 bytes tests/test_datasets/test_body3d_dataset.py | 8 ++++---- 6 files changed, 5 insertions(+), 9 deletions(-) delete mode 100644 tests/data/campus/actorsGT.mat create mode 100644 tests/data/campus/actorsGT.npy delete mode 100644 tests/data/shelf/actorsGT.mat create mode 100644 tests/data/shelf/actorsGT.npy diff --git a/mmpose/datasets/datasets/base/kpt_3d_mview_rgb_img_direct_dataset.py b/mmpose/datasets/datasets/base/kpt_3d_mview_rgb_img_direct_dataset.py index d59deb2b84..b965ba4c04 100644 --- a/mmpose/datasets/datasets/base/kpt_3d_mview_rgb_img_direct_dataset.py +++ b/mmpose/datasets/datasets/base/kpt_3d_mview_rgb_img_direct_dataset.py @@ -6,7 +6,6 @@ import json_tricks as json import numpy as np -from scipy.io import loadmat from torch.utils.data import Dataset from mmpose.datasets import DatasetInfo @@ -249,8 +248,5 @@ def _load_files(self): assert osp.exists(self.gt_pose_db_file), f'gt_pose_db_file ' \ f"{self.gt_pose_db_file} doesn't exist, please check again" - gt = loadmat(self.gt_pose_db_file) - self.gt_pose_db = np.array(np.array( - gt['actor3D'].tolist()).tolist()).squeeze() - + self.gt_pose_db = np.load(self.gt_pose_db_file) self.num_persons = len(self.gt_pose_db) diff --git a/tests/data/campus/actorsGT.mat b/tests/data/campus/actorsGT.mat deleted file mode 100644 index c73c8b0acac7fc0e0762302f5a691586501eec5f..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 4920 zcmeHK2~<RMgim%HwL=iPfw?%VtB zbDy)Vhm-AWy7^R7y0fjvlqJhoap|m8oUkQpc&n}GAz>!;IXo^Wj2lc}6Jka8SQbuq z=J4odrgXEJR%X+z%uVS`Qznymzd|}VYfv-uKOqR}N2pMh{LPcnTlWgT*_El5vp8%G zZ<-w;n?4wD8|C z1hEIBLrl*SI8>vRTCql22+v1w=d@IC^6oN(f3 zQ4xVFv#(tUzbXYoHM90d+5~!;)aWqlUxBzYLU7oVK<&%5k%3wS=2~9dqSsjo?4Ilq zWui8iopNr``HT)Qy3xS+GW8`G&C9J=A$%~YCv$A(7*;4OS+u|21)?F#^!ya%a93XKzcSq7aBPl3u z??`Q&K%&WT^UcK~0)>tW`fa)dj`CqNjxr!{KGj_* zvmVo<5=?_C?x;>{1AQmeMNJ7EAbpXnG@{`plw_XnRuQ*BL2c)wyWRpY>L0jpUA+$$ zXfYbbDT=^Re`YHyTnzup>;H6n_yD*jUUTJN?EydE=drtobwJ%nmUUpH1nA%IXwsy; zKMwqBJLZn_7eQAnH*w)|5pZ+N^B=19!J{oSSHbQ9&@9?`;K)faxE*QdId%=eDE)0@ z&$2GCb}E`w9MTQK)8ne$yL(~LOuNwuC;>@lBAZ6xaO0}?_g4K9c=4%2XIOqeq-inc z9TW7ySDs(xFlG(HU8pRs{#FWs=@XOFrjV$f?^_;dLSW3-uFez36WGPijouVZ;$+ch z*T?Xr(4EKId_IW8&Ci3^3w20j69T;uCV>+ZKuvc(flRlowp)fnP)#P@T$fGYu$zKH zM_+26t!yZw8Gi=`=BBDv{5F_bzc_7pPzU%q#H>qKdkH=vGtI{S&;|z?!NZe<)zI!c zebJC=A9N(_KgHA)!Fds8C%JowpRkl z8e=lkec#}4nbI1$F3I&uu1j)FlG|nb$Mmi7|EXj0aq?ECW3uug=KWpQMX0|UFWGyS zWA=N#V!p{&UKCjW3(hLT-aY>9IC6YhR7dV#j=P*kGW;obzvow3pGMV;%&h#p(xepr zv67NQ`5%>smHH*vy5uxe$`|~s=UiDm68cdss@n0q#Nl$U3Mgw zWRsp$&y})ylHOzm!6y=lEkqhoOoS5a2rj`P!ilYfKjHj3o+h<8c|2JwEANUYTjxWn zrmSzPBG0td9b#lk*RD`QA3=iaf^8<)d^1>MD8~qoaLohSeGG7Er;>T$PHR+d^zLxG zI7(o>;8NHNp&`E$QC5=&=)~=4rSm8zeh4geeNo9gd69 zo`i$DJQhY5O~RGBqdPj2Y>?PnwWyJ&AlS00CUks@F$Q;4M_m;%aGu72M6;sF$hzgH zG}g})-6!73dA?N(lYUOP9T3mJk)vKdnGrl5(R*>#9^hi=hGt?$J_n`SJPi}|{W1EB z((`&h`eBlANm$@P4z`2_4K^xs@w$HG!Lve7)O%tZJ*$O{?hP4w^Ap{1<&pB%Gi7d= z&_1TQG0qEJS58(xs}+C?ZQ5y}`&{wW$JQH6d$%x`jGxwALM+9|5N&Kr|0Lo zqlTyJVrQw%YC1|s@jcXajbLJ8Mq8nJ>gQOVGDDUAQM>P)1ZTvAjcY2Gxvp+heU1`5x!awGCbTnbk#b5nB1Io9^Z4)I=?@1mU1aoZnLkKGT>jdsO*Q9-wL zX20P>*&pN!c}|j_3uJooNBQ}`<(I$1x#Q!y{{1^!&IkE<;X|AgKl1iXUj4WqKg@T^ H`{>%QVREHa diff --git a/tests/data/campus/actorsGT.npy b/tests/data/campus/actorsGT.npy new file mode 100644 index 0000000000000000000000000000000000000000..c7be2a9f20326f768305e250c8ca6d622dcc27ff GIT binary patch literal 1136 zcmb7=`&W~70LI^mcI0BoMMW+$4H&j{4h)Gw+ViCYBtveBOD+Z|&gq~lX^L&1Ko~J5yb4+{^i^`)4 zIGJMUb_pkx%~_iz;PBX-ti6&PNk(?sUP-3-KOUK}TPn74>5hzjVr%vb;IRXF?0_{^ zTg_t^vj4yAf(~BRi4N#-Z+Cm@N4sayDEQ^*3QZTf8GhE0r3O?^pA6nwHV2WeZqfG! z75Iggi&Ze7=v+az|IZY8x8i16|Cefr&c|?`m6$Qt?OfS;hJo*JANRe=7s%OkKcGQm z#?w76E6aVXJhHKKCo_TC{U1y{_o6X+XnDkR5sQ5B(&pHc`#l&A$kRSPHUq(e3y*q> z=U~@W`WEgRz-+JiDK1|N;lRlbJ0nGU>dKP}y$#UXph_{T$DgY%ej}GtCMR4d60cZS;h@k#8CIJb7$DcyZJBmt5w8ulX#>S-jMTq!ApE)+wRN8@2($KQrpp&B>U#k{UC>Zluo*7F z|AfYmG5D_Liguw{1$pl=@n1X|vYc`8s(KdrwnsBDmZZcKC+yVeeKS}iE<66j*ZRN7 za_gak5v_#<6P*Gr=9j7~Z>(aGSZ8jZ+k69lxSkv9ZqkEi7j(GIKoOx=$Vhyy8SG1U zgnNQ57!=xgsXQt2d$VatX|4rLH%I3W$!Itwb8k3#QzW5eC_(l>hqhoJx8@WY(O7r4 zHEb5CDbI^e&r>6CQQ!Iq9}^PuwbIf`id=P!Uz?s!qetqqFb+n*#btb$OA+?9vVp}O z3}i!P5pG)zII;Oe1NZG&%0;;2W0^0WmGEh!!8Jc^`76-v+AzrbSoq57svW)z-Ei1AxV zk>u=<0N2|pY;2m82xlyqku?^4T)-mdxS4k4g9b7hxw9S&k{o7bXUHsK$zNZ0$SKJhvw0azVo|!l3&LaJBVMk38GtxrBmlhKm zB1*X;n@rWJT1SzEu3e(Q&uLUUb0b@oM$CJpJTFY4NcpSFYR;xG_!rM7 B361~& literal 0 HcmV?d00001 diff --git a/tests/data/shelf/actorsGT.mat b/tests/data/shelf/actorsGT.mat deleted file mode 100644 index e1530021bda024e4444ea4d11bc098c519d46ade..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 13968 zcmeI22T&AU*7rdWP*5Zo08zw%2_+~9h~*O74XMv?+wGXye`lBPAcC(GaJt*9xA$yTs@=MFr8e4 zKD7^6de8eFnnk{OL^<{@x@B^fFKbIBa=nT|lrXJA46~?llhh{&dy$yz46PdFdQ=j; zaHR${PwpqzoOpnWv$Y6sRIEiltU9-9$h{6}@C=%i-Sq@{xbISZ{Lo!w!(dBi9Y;2z zDH_sgv*spBDqmgVb)^-V85Wn(q4XnphT>I2OFkmfm)`SnQeGp(p5}4xm~>P!vIuhn zJAe!d@kkr1jw8~iqsJdcO(W;3k!T06X{6`koNDL57!rEa=1QaaAaYJr?SygeEYgeH zz|&gv3Hh`9u=)Vw*aulEVzNGHnH07VCC)f|N+g$~`5|1~ZVBb0`X)^h zZ4>Fprz2&$C(hnQ`2uJ`B*E}M><3PlgCj~0s?fN*dmKtJ4Ur=jP-W|7j%AJ;=pz={Pb&VzevJ29Yz^ue(!=XOWVf=e`7%{<3mkBYk9_^_QT6@9DE zp;Z621VxUpZ{MGLAJu;9ke;e?AGIuwX~*1ofN)+o(BHuE=# zp2C-C>!FK_?T4$-9;cJy&C@l=i3)-Vb;k?DC5kVgS@984;PUXA?zebkPqE(8n87PZ ztxB}g>)tdpbJKxE!ueKY@%dW!3j2P*W~XN)=JGH_00V% z-S5t@xHQbSqj(y5xiW0CanA%|oUOX}(qkOif4i`OfJ`FmIWE^Kf1X0Blv@ra9-c)u zuPZRqZjeC`}?hM&m@;%M%FyE0iMIQn0|tjVa`S^8+-`eG1vIJtEb& zwhOh7=GKkHbfRBJVtWUQ-l7W!1v+&mKcI2q_P3AaccBGh^R%Xt9#puodd#J_34P)x zS1hu-1$DQ++RJ777F9mDFaDx=4;mwGV?tW&L#JsmZsrm1(ecG{jO@pDG-GHrXUvT^ zs88Yz+p~V*=)L9L8azg?5#jdwZ-ll1Bzbps`LTroB(leD*Ie6MBzG>DQ%{G1yiPM! zeDGon;j{_#sI;6!8j4c;LUA*Q$a$R8V)_i?Ro37)lRJS#FFW@brH>-R=Ot3}R?Z<$ zb9C{G{=d$J-}+%nxo}IH(KIraliQ(PIDu&C<*z!;97pJ7A%`0JCz08i$b7<+DTJ7K z)l---gU|?6N%rg+WCt$Z+`4@Nq4O$yO2the@fPC@EB0}uZ~vAnhB{M79F^z6ec|8x zVeet_de3WbQ9q~lX=`Q|x?1XDn!0#5T3WG?@wu=MHC1g-VU9JS?rSQ(1#W3ZbCNLy zeClseX8lGE8I2xv{j-dPxLy4y4XI!3*w%^WRu>JVHngLGCu3!eF8>Oj%GgF9u^|tabsTpLk+C0#8#SC&if0vJ4 z)C5AhS-HAGU;@#XZ``f7e+*gSV8M`NpF(<5EpMc}ocwz{xVI&wm+k9BZOX31iWA?W zT~&kEl`Xr_E1GjpIbss*{(bAGM>(K@u|e$Ups{R*^q*g!vlTsyp878%NRrJM&az4oR8U8+%Fls~`Ta^7-PZKJNdOuToga?4RYg@9+7Z>we^`?|Et!>*7cM-{!aP&zJl||L^&&lXcEM9Ip&v zU|~J?Tv;I26W$)Oure|71SEXpj(4e85M$V#VbE}3sH$0dYKc48@}^`;^b-Jor2M?t zCp;_@*?lZa!3&aZYCcIFa{=~Qp%FF?SID^PInO?ggU$_FuXzn{pmI}n;<}9&WJ$KU z_rDGU0nfYGwD(ka5mL!^%p(xyB}~~O_XNOk(+(TA-BhUEsQ9f^I}~1@xIsP9<`2)x zhdf%o5nw^gyu_l@4W_Cp_wD$E0ka+FmJhNKLC``B`w8)bL1%$-X+H;WnH7ukJ!%HO zY8U3Q!wZ{BX>eWdQ2Zc~2A*NTuaj^zaIJ|rAGD7G_k7IEzY0=dAUHcdp^*&tHDWet zyV7CLm$xHEx9PB6YNd1-HysSmu2$fVA%l8$?9xCDGR$fhTjILO5RVUV5K*FnxdOWH zl`0kF6ZckKZKgw3`NIC&A!KmIQ3#Hb{&4DiU)iy^cHYFOCrDpw+;mjXDnQ9edwf>kA;&AgN^fj1UNZVW#g>m3)0z(L!~CJkRDrp zCBoJ2m+i7Pv+uHg1q0&Th%OvS3`pX7%DsIV6I$hh=cPMo(0ol+H&~GdHW_Sgh>fhhEPs>+GRj2jRW%G~ zI}jM!Y{q~w*N-2bxKiQ7UGoaNnv@^e(g zYr6y*$gk`9rpQ4BTB1!Lwwel3(s{$k1{$;li{-s~PXiuBTh1d*G+3Lo@11KH6`Djl z7CDyF;FahJ(N{o)YssOq^BELyHH*0LB03OWWo&=-IE)02>`p3en|wft^h_N!2nIb5 zBiC5gd;4vJ1Rq!sp@c_5Db&ylB#nDcyy=lGdY=<&T^RyV9o|x2^G)f3Y{7?n?<4IYTj*QA;Sp|)I|%Vb z+#?dbz&TQSX6&T}4CD;OhCR~!i(URFa=Iv9>En4h0_Zs(9^Ti9hs43ATK`l}ctz^% zN}YEGjp;~}b8Id^cpfL_K7<9da*@1V9}8~L3dw`Lo`0CrykGPWhOh-dUm8Z4UzZ3m z>BBSQNpA42QS#23N({tIJT)W?5MbM#ua#?d`avHKMGzGSP_K@ReP3hx%k%4xa@yx) z*5*EYI!JBoniv|T!~LX_OKeurA)8@(Yk3kG2&OZ_dHQ5v`mzb~4U>WUZprFU4F}MHE<8=Cwm&uy%8BkPSHb1cre zjc3`-QDOKqOD%6bA0b1@-v=Xt~1v45P?>m}0bgBwXua)82N8tn~ZOq_O55EfFy z3%*EKdciA?n(hngK5(a;x%R{pUl13w|Ipgt250)(SDwlJ^EiEf@{3kB%kS)39)$se zRIt*|oQO4}K|uA3WXrX501=BS4t_cmUG5F%dqM>}HGZ~9H5$Az`OI_6?~il(u=A!WLRjwhc?P zVqhUJt@x?77bH{)2bX17z_o7Gp6Z%EkJG2?$#{$u4y0|rE~t*co*>Xc}N8wG-_X>b1c8JkF+(7esh79?t)yp4_!c< z>%g`9EL}b@vfs`91{O4nm4@^cJt3&hYP@MO49cYkYtw{i@Dhv1pS=?Z5?2nKBSZzj zvm0ebDMml_vt4aGr}9M+5<1qU6oh(_%Cz1%F=G{=o=agkI;{Aentc1kJFq-S#3#}qnFB63Op6T z@ho-zsiz<4UOvAjpANa!-YFcq46r^hue~~s4xQ|tHd5cn!0UMn!<|Y7fnYC=)ov8H z=v#Bj^Ewqm^t(D4QB;`Z-yZk)Jso(Z#M*bT^mpb<{+D{H{!r_#Q*rKtAB+iB?7J2C zQ%_fKdy$*v;0Hoe!=7EsiGVG(EErnELc=pTqk?@nxMyeN&?3W%OXpc4z1kNv6!yzB zRJ%Zwwc`-KjLk3Gr6NaU*uR*}djDvnH8ig;L-8PcKO1zWsG{i&z7 zWun`CgqT3(91{wO3x(UO)9BKC3p6<5vqNAN8yzA8G>O)URIr@bU^{L|gG9M%pLy0g zRFLbv64xCHD8^Thd)WtqS>N_GS4bpSEYiVLN%%l&)WNpm9Y6K-C7Xmv#~dO=p61W5 z4)=nGS2H>^om@dtw9Qg=r5gmC-gNwC79N(oPb$ngOai~7F_MCGdpMV@(&1$G>v|{3 z)$GdU?KEh75M;Yoo(94<3*KqT(g2rAj7*)UgF=5>vyAXhJxzIDSJ}3U3Ot`rxqTa_ zfXxl{EQ#dXzx2<0gJN2mIdwbmpJ-fK>1O} z_)*{Z|JC*3AL<^z?f?Dyw{?>LE&Bj}_TTv*JN|Y5nl)wEzSoU@)&BpnDW-%(``i5>}BQC(fgU#x){(LF=YN0m_VTkvdcFy;4x3a1Jz+BG;lq-B=C`y zy>i-yxmoKgBc36zAP**NQ@Y-kq#Fte?!tp|O`#AecBSH(ECWtOX{O1r^m9w7`7*T& z4A4tIWD@v^4tIhEM1t2*pn5{vEqJpZ%qm=Em!kOrZ_K@;xpEY+l9zT94QD{FimQr* z0TJvfT%#=2@X(XZtufx@4j#!C{7k?=gaDBdu+tm#dvZxRjwCoWMwG|wCP^}T>jSu0+vNA9K_E&}Vy4}vKm~W+ zQtrJ(Xq(O5pUUqIBL3Uf+BANx66GJ*`z&COwE z>&S4LecR@p%>i)6cz=zrmOn5$O9j@j>f#Y!TCdGIl0hvbx9bf{{~p#0%@5s6h81oC z7CiAmAYXQ0t4}5f2sbkou8osmNz0+}kF9u6adhfb7sbFCyUTeZIvDu$%zBxDn>XAj znrpX=_kr=Sg9}BiPH>xXSbTMu4U98H5@@fCfpYZVt}z#5(C3fS%2cz3qfIJKN=z3} z?ovK_CRrcUju`K;)7baRb|GyO*IUYofu1_k7Yu7R&=qjw#s;{-mZGR4Vb)&LMCkbv zjaPV(lZjON67L0aCj@;$4q%|mFKteg&k{uX@eVt+|8K$e2)U(;;zqm=LLbz>9YlH?j+E+k<~5BBEs#L zxFt3p0-(Qso}s#i0w+BgDirXB7wl@cLn1t3Qu$zZOPMR&B(#Nzg}K0tg~g2^XHQ7& z)je9F=MAJf{*lQgfL49G}4l!c?s6VtF^vyLX z@&&d;TE3n#tA19bJu#Q;1NJ8`7B4>%1j5iaD*PY-34k%{hjUTOFy`Y+)-*-5&&(OX66=_UaZNo$dP(RhEs0QeP_!^@NSh{ zG9{4+jt=Q^goq$`Q9Y>^<3R(R`GZU6roG_)2f>?jTpp0V^;C{@v>gNrNvT+e+XAKW zR_$`E2OxrWXErRwgH8Xq?Zyul;L)&)Ir>@sFLwDe>ZQNs^yW0#4tzNSS|rutv@DrG zntq=6{3`?2EqStYcUdUQ9))GCt#t6x6rM2fWq_T6lqQ!x6ACU~3h&t(3R!aIj@af< z5N6BetXt3GQNiukJp&9Vm;dUNjr~2RZ9Y0)?hy2Yth0Hkg3*2;952zbR-OV6Cnm#3 zx*1R+bhAQ0l?Y{>;a?5T<6*bpkv&%T+~M`PBkMBOVxZyO_PNwfqNe}*IQ>vd?}PO)9frE^u76^{gdShl9R_qJ6xD3V$Q25O{UHu3Q&@b) zB45v5J3@yXh3$;|;FYB&pm|M0k9|O?TeQ16Db6>@nv2_i@^~>8VJ2R3H>; zzuLWO8ySWrq$dPo$#4n(-luLW8N!|RG?gp_K&`CwXGl=l7hxBp2Wsql5vw(06*b4`SfaTZzR~ z7FTi)&0#j5^Z_x_DH+2zPC%KE3^TfJ3w85Y-SI&-? zzIehjH^g>t76yXPd|HR+bq4JUAq|N}M<@;M5ap1^0b{S;wN4h_?@IWUM(N-ISGoC` z73=?9mlToFF!_51bU2_3psmv!h_F+eL|t6wpI&olBo!Q>~WH>oVIyeCEhHTg)NQ_K87oxQcsJID`u zL`6hDFe%U;q}iBL&w%R{bW~1{2)XCZ4IU=r!QiuKdkxba^i>0#t7qN7ffU%0=ZA;M z*{7mD8;Bsu`5~aB)fvc}mehx_^5L(3fAB+3%UT3<-(cmVUavxjZ>;?)tv6>)gG^ZK zjwpfYnO973d#7nNEfESiFdS-V#M(!{(9$EMO^1t*ZFLRa(xFl-K_sby2}U~mdz&et z;9SGdT|dEqFK>!zJ_ZaBtf!1>%>Lx`?gu?=VKYJC*0Z_iZHW(19W+D3YJFfNaqDb3 zZ!ipa6n|U(oB|KD=3|fTX3e`W8m;UAOHbdy4ku`Og7vPaEyNrgWacA1Uq!v4|BSAj zat{GiOqT8MKIH-JR-&ZV5msHBwZ8t9?^S;2>FB1z&Zj;GLe0C4Ynt8#!v5VihN5iA zz%HjY5Kuw}5hwN7@LDpcIw?peZw`dbfh|Q_C;Wl5btG)-PypQWbW{(!NrsQbyt)CJ z6xh@d!ee=h3=&!bAD^uahWr#e*XEp`oK9>u6XfM&8OoyZqp^u;Oj+$!HAR-dfr1YUK{ttGLTgS^L&9+nxQE zUB-j@>(d-(Sbj0AE*DHa><%BRxvq&nc7{x9Aw|y68I0AC*1OF(823x7j2Z9*v!G{o z9eH?=OnAmf@5O-now-8S%b36GX<=B*K9dv#cUl(?jVqF1(`Bl*wjU7$L@>FAysYmM zA~G91>&Vc%tZH6qzc+-N?Re9{${QSPl$V}z%Oho-vl ztKV4P>GBSxY$>}o7XUcTI&>vpvN&3xoUMVbR9gfOSP2> ziulCqw;H{mwEDV9hoJ{dzlD2^N_LPQlr*WLV+*@CYM;lv#lXr?`{Tm|);D;KzU=lF kEC99DkBntM{FnLipL*%PRpF diff --git a/tests/data/shelf/actorsGT.npy b/tests/data/shelf/actorsGT.npy new file mode 100644 index 0000000000000000000000000000000000000000..45748c90e46604886166fd24acea84bd324316b5 GIT binary patch literal 2144 zcmd6n=|2?+7snYyjkKVmZZ4TF(F|qF&}os0hm38gsVGZ#a&zsHB}=pn4OuEBiV~3} zT!a&4tC(D5yXw;ATGHJ~<$64Sz`U4u&v|n`=lAM-zMt>smwL$A_SeIb5+M@dBu}qE zj{uS>k!0p=LeeFYyr}^}0VjOjr~#f{|H~~;Py)S{@<8$lKd+@*Ylkjzr!G<7U}pWy%7}Fr-1b)<3Hs z*8Sd~AiGeBMbY;yOw&tH^pT)v6~l&}^Aa?*pKgLCe^$Oo^94MyCp1rgHvzXF*Zvkf zUJUn|V}gm!&9MFYdwJ=YK1e$DsH4Nc5ABO$Gx)J>5X|c3%BB^dv4efDwPh|y(l+Xg z;#yE6h31p%I|<`gQw|fTrMRrIHfkZO6{i&CwmX`0;nbn@1Rk*(U9#EHt*-UB{b(xp z*)<`|yEU}c6qX@D;;oXJaR(%}K&q#|5O`*rp4XY=zfA_JPG=|UPdN`xBmhhp14Rjp`gY0h)K>9BeHeIR_+9i$V(yjj4 zPo;G?2{VfpK;gYA4s2}4qKV~w(Ww*AeO+;_DyJSzt~Qe8sydOKBW3xdl>^CYuhu<( zhS0ci<%4ikena46MLP^0T|vi#LE3j$gIk z&VuS6xg%Q3^6txZV|>`M zq4GBy`B|X2-5FcvJcx8!-uOkT2%O*VyLez1A56^*ZX~;~P$h}#EiMp2r~@}x{}vC@ z2<}cH;&Jq^qlbygd9dnlvJN+e3&Oax(=J!vW9st6=Xl{0INy$1Gm|R-uDHQRe}aJ@ zlZbgw6z0I>R6RbS2%vCUSvoCu0NYEg1H|X%|20p6X_e-XgAjh&Pw4WyJBSyJG)``l zm;z2|`S8Ii9)x)_gl-9=cytxNi#<09zt~i)KAOMUl!~A>0XUm8auPT_n5-DRUt_7?f-~>(77XCoitf?_o!i$zeH?NcegZxBz{e5V&4Dgtfl4Hq{C4-&f6r1TE3P|ebd zCMXlZh3}2#^Qb)7^s;avLx+t#1@+LdB@b>K=EfhFG}@I187vx`MfpipLwbW%F3=)U;2OO zWLBouO$E zBPqp+y(|DjW!{mV(jM%3m_iNg;(%?-)`m+Y5sVf&hLouG;z*33nWV`Faf?gDA1jwQ z4cKq}d-Wi0Nc7t?6Dfjw6*{3Oukj$Nzn6W|h=pw`(!PhPMG$iTg>usx9_UR63}3Qk zBXirgBAzZEYquQ`5#HtD~le$oX<9#&X{@lFM(l2%>V!Z literal 0 HcmV?d00001 diff --git a/tests/test_datasets/test_body3d_dataset.py b/tests/test_datasets/test_body3d_dataset.py index 1b41c20e21..f2337c592b 100644 --- a/tests/test_datasets/test_body3d_dataset.py +++ b/tests/test_datasets/test_body3d_dataset.py @@ -379,7 +379,7 @@ def test_body3dmview_direct_campus_dataset(): cam_file=f'{data_root}/calibration_campus.json', train_pose_db_file=f'{data_root}/panoptic_training_pose.pkl', test_pose_db_file=f'{data_root}/pred_campus_maskrcnn_hrnet_coco.pkl', - gt_pose_db_file=f'{data_root}/actorsGT.mat', + gt_pose_db_file=f'{data_root}/actorsGT.npy', ) test_data_cfg = dict( @@ -398,7 +398,7 @@ def test_body3dmview_direct_campus_dataset(): cam_file=f'{data_root}/calibration_campus.json', train_pose_db_file=f'{data_root}/panoptic_training_pose.pkl', test_pose_db_file=f'{data_root}/pred_campus_maskrcnn_hrnet_coco.pkl', - gt_pose_db_file=f'{data_root}/actorsGT.mat', + gt_pose_db_file=f'{data_root}/actorsGT.npy', ) # test when dataset_info is None @@ -507,7 +507,7 @@ def test_body3dmview_direct_shelf_dataset(): cam_file=f'{data_root}/calibration_shelf.json', train_pose_db_file=f'{data_root}/panoptic_training_pose.pkl', test_pose_db_file=f'{data_root}/pred_shelf_maskrcnn_hrnet_coco.pkl', - gt_pose_db_file=f'{data_root}/actorsGT.mat', + gt_pose_db_file=f'{data_root}/actorsGT.npy', ) test_data_cfg = dict( @@ -526,7 +526,7 @@ def test_body3dmview_direct_shelf_dataset(): cam_file=f'{data_root}/calibration_shelf.json', train_pose_db_file=f'{data_root}/panoptic_training_pose.pkl', test_pose_db_file=f'{data_root}/pred_shelf_maskrcnn_hrnet_coco.pkl', - gt_pose_db_file=f'{data_root}/actorsGT.mat', + gt_pose_db_file=f'{data_root}/actorsGT.npy', ) # test when dataset_info is None