Skip to content

Commit

Permalink
Add custom dataset notebook (#1592)
Browse files Browse the repository at this point in the history
* add

* change pip install

* upgrade version

* remove pprint and minor style changes

* fix name and fix link issue

* remove pretyformater import

---------

Co-authored-by: Eugene Khvedchenya <ekhvedchenya@gmail.com>
Co-authored-by: Shay Aharon <80472096+shaydeci@users.noreply.github.com>
  • Loading branch information
3 people committed Nov 14, 2023
1 parent 58ec6ae commit 4931566
Show file tree
Hide file tree
Showing 3 changed files with 1,463 additions and 1 deletion.
1 change: 1 addition & 0 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@ NOTEBOOKS_TO_RUN += src/super_gradients/examples/model_export/models_export_pose
NOTEBOOKS_TO_RUN += notebooks/what_are_recipes_and_how_to_use.ipynb
NOTEBOOKS_TO_RUN += notebooks/transfer_learning_classification.ipynb
NOTEBOOKS_TO_RUN += notebooks/how_to_use_knowledge_distillation_for_classification.ipynb
NOTEBOOKS_TO_RUN += notebooks/detection_how_to_connect_custom_dataset.ipynb
NOTEBOOKS_TO_RUN += notebooks/PTQ_and_QAT_for_classification.ipynb
NOTEBOOKS_TO_RUN += notebooks/quickstart_segmentation.ipynb
NOTEBOOKS_TO_RUN += notebooks/segmentation_connect_custom_dataset.ipynb
Expand Down
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -224,7 +224,7 @@ model = models.get("model-name", pretrained_weights="pretrained-model-name")
### Object Detection

* [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://bit.ly/3SkMohx) [Object Detection Transfer Learning](https://bit.ly/3SkMohx)
* [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://bit.ly/3dqDlg3) [How to Connect Custom Dataset](https://bit.ly/3SkMohx)
* [![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/detection_how_to_connect_custom_dataset.ipynb) [How to Connect Custom Dataset](https://colab.research.google.com/github/Deci-AI/super-gradients/blob/master/notebooks/detection_how_to_connect_custom_dataset.ipynb)

### How to Predict Using Pre-trained Model

Expand Down
1,461 changes: 1,461 additions & 0 deletions notebooks/detection_how_to_connect_custom_dataset.ipynb

Large diffs are not rendered by default.

0 comments on commit 4931566

Please sign in to comment.