Skip to content

Commit

Permalink
Merge branch 'refs/heads/feature/SG-1488-YoloNAS-R-integration' into …
Browse files Browse the repository at this point in the history
…feature/SG-1448-OBB

# Conflicts:
#	src/super_gradients/training/processing/__init__.py
#	src/super_gradients/training/processing/defaults.py
#	src/super_gradients/training/processing/processing.py
  • Loading branch information
BloodAxe committed May 17, 2024
2 parents 03c876b + 86cbee3 commit 5729722
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 4 deletions.
12 changes: 12 additions & 0 deletions src/super_gradients/training/processing/processing.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
import cv2
import numpy as np
from super_gradients.common.abstractions.abstract_logger import get_logger
from super_gradients.common.deprecate import deprecated
from super_gradients.common.object_names import Processings
from super_gradients.common.registry.registry import register_processing
from super_gradients.training.transforms.utils import (
Expand Down Expand Up @@ -907,3 +908,14 @@ def get_equivalent_photometric_module(self) -> Optional[nn.Module]:
@property
def resizes_image(self) -> bool:
return True


@deprecated(
reason="This method is deprecated. Please use `super_gradients.training.processing.get_pretrained_processing_params` instead.",
deprecated_since="3.8.0",
removed_from="4.0.0",
)
def get_pretrained_processing_params(model_name: str, pretrained_weights: str):
from .defaults import get_pretrained_processing_params as _get_pretrained_processing_params

return _get_pretrained_processing_params(model_name, pretrained_weights)
10 changes: 6 additions & 4 deletions tests/unit_tests/export_pose_estimation_model_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,15 +17,17 @@
from super_gradients.training import models
from super_gradients.training.dataloaders import coco2017_val # noqa
from super_gradients.training.models.pose_estimation_models.yolo_nas_pose.yolo_nas_pose_variants import YoloNASPoseDecodingModule
from super_gradients.training.processing.processing import (
default_yolo_nas_pose_coco_processing_params,
from super_gradients.training.processing import (
ComposeProcessing,
ReverseImageChannels,
KeypointsLongestMaxSizeRescale,
KeypointsBottomRightPadding,
StandardizeImage,
ImagePermute,
)
from super_gradients.training.processing.defaults import (
default_yolo_nas_pose_coco_processing_params,
KeypointsLongestMaxSizeRescale,
KeypointsBottomRightPadding,
)
from super_gradients.training.utils.media.image import load_image
from super_gradients.training.utils.visualization.pose_estimation import PoseVisualization

Expand Down

0 comments on commit 5729722

Please sign in to comment.