Skip to content

Commit

Permalink
Adding pose estimation to the readme (#1375)
Browse files Browse the repository at this point in the history
* Update readme

* Fix small bug in __repr__ implementation of KeypointsImageToTensor

* Update pose estimation image

* Fix link
  • Loading branch information
BloodAxe committed Aug 15, 2023
1 parent 67f7a4e commit ca8fcbf
Show file tree
Hide file tree
Showing 2 changed files with 33 additions and 2 deletions.
35 changes: 33 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,7 @@ ________________________________________________________________________________
<img src="https://github.com/Deci-AI/super-gradients/raw/master/documentation/assets/SG_img/Segmentation 1500x900 .png" width="250px">
<img src="https://github.com/Deci-AI/super-gradients/raw/master/documentation/assets/SG_img/Object detection 1500X900.png" width="250px">
<img src="https://github.com/Deci-AI/super-gradients/raw/master/documentation/assets/SG_img/Classification 1500x900.png" width="250px">
<img src="https://github.com/Deci-AI/super-gradients/raw/master/documentation/assets/SG_img/PoseEstimation.jpg" width="250px">
</div>


Expand Down Expand Up @@ -121,6 +122,16 @@ pip install super-gradients
```

## What's New

__________________________________________________________________________________________________________
Version 3.1.3 (July 19, 2023)

* [Pose Estimation Task Support](https://docs.deci.ai/super-gradients/documentation/source/PoseEstimation.html) - Check out fine-tuning [notebook example](https://colab.research.google.com/drive/1NMGzx8NdycIZqnRlZKJZrIOqyj0MFzJE#scrollTo=3UZJqTehg0On)
* Pre-trained modified [DEKR](https://github.com/Deci-AI/super-gradients/blob/master/src/super_gradients/recipes/coco2017_pose_dekr_w32_no_dc.yaml) model for pose estimation (TensorRT-compatible)
* Support for Python 3.10
* Support for torch.compile
* Other bugfixes & minor improvements. Check out [release notes](https://github.com/Deci-AI/super-gradients/releases/tag/3.1.3)

__________________________________________________________________________________________________________
30th of May
* [Quantization Aware Training YoloNAS on Custom Dataset](https://bit.ly/3MIKdTy)
Expand Down Expand Up @@ -232,6 +243,19 @@ model = models.get("model-name", pretrained_weights="pretrained-model-name")
</br></br>


### Pose Estimation

#### Transfer Learning

<table class="tfo-notebook-buttons" align="left">
<td width="500">
<a target="_blank" href="https://colab.research.google.com/drive/1NMGzx8NdycIZqnRlZKJZrIOqyj0MFzJE#scrollTo=3UZJqTehg0On">
<img src="./documentation/assets/SG_img/colab_logo.png" /> Pose Estimation Transfer Learning
</a>
</td>
</table>
</br></br>


### Object Detection

Expand Down Expand Up @@ -637,7 +661,12 @@ Detailed list can be found [here](http://bit.ly/3GnJwgZ)
- [DarkNet-53](https://github.com/Deci-AI/super-gradients/blob/master/src/super_gradients/training/models/detection_models/darknet53.py)
- [SSD (Single Shot Detector)](https://github.com/Deci-AI/super-gradients/blob/master/src/super_gradients/training/models/detection_models/ssd.py)
- [YOLOX](https://github.com/Deci-AI/super-gradients/blob/master/src/super_gradients/training/models/detection_models/yolox.py)



### Pose Estimation

- [DEKR-W32-NO-DC](https://github.com/Deci-AI/super-gradients/blob/master/src/super_gradients/training/models/pose_estimation_models/dekr_hrnet.py)



__________________________________________________________________________________________________________
Expand Down Expand Up @@ -667,7 +696,9 @@ Deci provides implementation for various datasets. If you need to download any o
- [Coco](https://github.com/Deci-AI/super-gradients/blob/master/src/super_gradients/training/datasets/detection_datasets/coco_detection.py)
- [PascalVOC 2007 & 2012](https://github.com/Deci-AI/super-gradients/blob/master/src/super_gradients/training/datasets/detection_datasets/pascal_voc_detection.py)


### Pose Estimation

- [COCO](https://github.com/Deci-AI/super-gradients/blob/cadcfdd64e7808d21cccddbfaeb26acb8267699b/src/super_gradients/recipes/dataset_params/coco_pose_estimation_dekr_dataset_params.yaml)

__________________________________________________________________________________________________________

Expand Down
Binary file added documentation/assets/SG_img/PoseEstimation.jpg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit ca8fcbf

Please sign in to comment.