Update version to 0.0.1 and rename aggregation method to 'aggregation_RPEA_joint_level'

Author: xiu-cs

fmpose3d/__init__.py

@@ -7,15 +7,15 @@
 Licensed under Apache 2.0
 """
 
-__version__ = "0.0.5"
+__version__ = "0.0.1"
 __author__ = "Ti Wang, Xiaohang Yu, Mackenzie Weygandt Mathis"
 __license__ = "Apache 2.0"
 
 # Import key components for easy access
 from .aggregation_methods import (
     average_aggregation,
     aggregation_select_single_best_hypothesis_by_2D_error,
-    aggregation_RPEA_weighted_by_2D_error,
+    aggregation_RPEA_joint_level,
 )
 
 # Import 2D pose detection utilities
@@ -27,7 +27,7 @@
     # Aggregation methods
     "average_aggregation",
     "aggregation_select_single_best_hypothesis_by_2D_error",
-    "aggregation_RPEA_weighted_by_2D_error",
+    "aggregation_RPEA_joint_level",
     # 2D pose detection
     "gen_video_kpts",
     "h36m_coco_format",

fmpose3d/aggregation_methods.py

@@ -87,7 +87,7 @@ def aggregation_select_single_best_hypothesis_by_2D_error(args,
     return agg
 
 
-def aggregation_RPEA_weighted_by_2D_error(
+def aggregation_RPEA_joint_level(
     args, list_hypothesis, batch_cam, input_2D, gt_3D, topk=3
 ):
     """

scripts/FMPose3D_main.py

@@ -12,7 +12,7 @@
 from fmpose3d.common import opts, Human36mDataset, Fusion
 from fmpose3d.common.utils import *
 
-from fmpose3d.aggregation_methods import aggregation_RPEA_weighted_by_2D_error
+from fmpose3d.aggregation_methods import aggregation_RPEA_joint_level
 
 args = opts().parse()
 os.environ["CUDA_VISIBLE_DEVICES"] = args.gpu
@@ -121,7 +121,7 @@ def euler_sample(x2d, y_local, steps):
                 output_3D_s[:, :, 0, :] = 0
                 list_hypothesis.append(output_3D_s)
 
-            output_3D_s = aggregation_RPEA_weighted_by_2D_error(args,
+            output_3D_s = aggregation_RPEA_joint_level(args,
                 list_hypothesis, batch_cam, input_2D_nonflip, gt_3D, args.topk
             )