Add pose model (#1593)

* all code squashed

* add to Makefile and update Readme

* remove ppritn and fix pip install

* update with all cell outputs

* Feature/sg 1277 depth estimation dataset support (#1705)

* added sample, datasets and transforms

* added docsand instructions to dataset

* dataset examples and imports

* added plot

* pipeline adaptor for albu added

* added albu test

* modified object names to be more clear

* removed apply transform

* added depth visualization

* minor improvements for visualization module

* fixed getitem returning sample

* fixed getitem return type

* fixed getitem redundant clipping

* fixed plot ignore

* fixed unit test

* added clarifications in getitem and load_sample

* updated plot()

* updated casting image to fp32

* update notebook to 3.5

---------

Co-authored-by: Eugene Khvedchenya <ekhvedchenya@gmail.com>
Co-authored-by: Shay Aharon <80472096+shaydeci@users.noreply.github.com>

Makefile

@@ -44,6 +44,7 @@ NOTEBOOKS_TO_RUN += notebooks/transfer_learning_semantic_segmentation.ipynb
 NOTEBOOKS_TO_RUN += notebooks/detection_transfer_learning.ipynb
 NOTEBOOKS_TO_RUN += notebooks/how_to_run_model_predict.ipynb
 NOTEBOOKS_TO_RUN += notebooks/yolo_nas_custom_dataset_fine_tuning_with_qat.ipynb
+NOTEBOOKS_TO_RUN += notebooks/DEKR_PoseEstimationFineTuning.ipynb
 
 # If there are additional notebooks that must not be executed, but still should be checked for version match, add them here
 NOTEBOOKS_TO_CHECK := $(NOTEBOOKS_TO_RUN)

README.md

@@ -218,7 +218,7 @@ model = models.get("model-name", pretrained_weights="pretrained-model-name")
 
 ### Pose Estimation
 
-* [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1NMGzx8NdycIZqnRlZKJZrIOqyj0MFzJE#scrollTo=3UZJqTehg0On) [Pose Estimation Transfer Learning](https://colab.research.google.com/drive/1NMGzx8NdycIZqnRlZKJZrIOqyj0MFzJE#scrollTo=3UZJqTehg0On)
+* [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/Deci-AI/super-gradients/blob/master/notebooks/DEKR_PoseEstimationFineTuning.ipynb) [Pose Estimation Transfer Learning](https://colab.research.google.com/github/Deci-AI/super-gradients/blob/master/notebooks/DEKR_PoseEstimationFineTuning.ipynb)
 
 
 ###  Object Detection