Actualize pose estimatino zoo. (#1114)

* Actualize pose estimatino zoo. Remove old models that are not supported and we don't provide recipe & weights for them * Actualize pose estimatino zoo. Remove old models that are not supported and we don't provide recipe & weights for them --------- Co-authored-by: Shay Aharon <80472096+shaydeci@users.noreply.github.com>
Deci-AI · Jun 5, 2023 · f46e756 · f46e756
1 parent b4eec92
commit f46e756
Show file tree

Hide file tree

Showing 7 changed files with 34 additions and 111 deletions.
diff --git a/documentation/source/model_zoo.md b/documentation/source/model_zoo.md
diff --git a/src/super_gradients/common/object_names.py b/src/super_gradients/common/object_names.py
@@ -302,8 +302,6 @@ class Models:
 
     DEKR_CUSTOM = "dekr_custom"
     DEKR_W32_NO_DC = "dekr_w32_no_dc"
-    POSE_PP_YOLO_L = "pose_ppyolo_l"
-    POSE_DDRNET_39 = "pose_ddrnet39"
     YOLO_NAS_S = "yolo_nas_s"
     YOLO_NAS_M = "yolo_nas_m"
     YOLO_NAS_L = "yolo_nas_l"

diff --git a/src/super_gradients/recipes/arch_params/pose_ddrnet39_arch_params.yaml b/src/super_gradients/recipes/arch_params/pose_ddrnet39_arch_params.yaml
diff --git a/src/super_gradients/training/models/__init__.py b/src/super_gradients/training/models/__init__.py
@@ -117,8 +117,6 @@
 from super_gradients.training.models.segmentation_models.ddrnet_backbones import DDRNet39Backbone
 
 # Pose estimation
-from super_gradients.training.models.pose_estimation_models.pose_ppyolo import PosePPYoloL
-from super_gradients.training.models.pose_estimation_models.pose_ddrnet39 import PoseDDRNet39
 from super_gradients.training.models.pose_estimation_models.dekr_hrnet import DEKRPoseEstimationModel, DEKRW32NODC
 
 # KD
@@ -328,8 +326,6 @@ def inner(*args, **kwargs):
     "STDC2Classification",
     "STDCSegmentationBase",
     "CustomSTDCSegmentation",
-    "PosePPYoloL",
-    "PoseDDRNet39",
     "DEKRPoseEstimationModel",
     "DEKRW32NODC",
     "KDModule",

diff --git a/src/super_gradients/training/models/pose_estimation_models/__init__.py b/src/super_gradients/training/models/pose_estimation_models/__init__.py
@@ -1,5 +1,4 @@
 from .rescoring_net import PoseRescoringNet
 from .dekr_hrnet import DEKRPoseEstimationModel, DEKRW32NODC
-from .pose_ddrnet39 import PoseDDRNet39
 
-__all__ = ["PoseRescoringNet", "DEKRPoseEstimationModel", "DEKRW32NODC", "PoseDDRNet39"]
+__all__ = ["PoseRescoringNet", "DEKRPoseEstimationModel", "DEKRW32NODC"]
diff --git a/src/super_gradients/training/models/pose_estimation_models/pose_ddrnet39.py b/src/super_gradients/training/models/pose_estimation_models/pose_ddrnet39.py
diff --git a/src/super_gradients/training/models/pose_estimation_models/pose_ppyolo.py b/src/super_gradients/training/models/pose_estimation_models/pose_ppyolo.py