Skip to content

Commit

Permalink
Refactor Keras Feature Notebooks (#1238)
Browse files Browse the repository at this point in the history
* Refactored feature notebooks in Keras. Standardized function usage across notebooks. Updated model examples to use the ImageNet validation set as the representative dataset (except for the pruning and qat notebooks). Added keras post training quantization notebook

---------

Co-authored-by: liord <[email protected]>
  • Loading branch information
lior-dikstein and liord authored Oct 8, 2024
1 parent b26dd82 commit fb13d32
Show file tree
Hide file tree
Showing 15 changed files with 3,587 additions and 3,542 deletions.
12 changes: 6 additions & 6 deletions tutorials/notebooks/mct_features_notebooks/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,12 +10,12 @@ These techniques are essential for further optimizing models and achieving super
<details id="keras-ptq">
<summary>Post-Training Quantization (PTQ)</summary>

| Tutorial | Included Features |
|------------------------------|-----------------------------------------------------------------------------------------------------|
| [MobileNetV2](../imx500_notebooks/keras/example_keras_mobilenetv2_for_imx500.ipynb) | &#x2705; PTQ |
| [Mixed-Precision MobileNetV2](keras/example_keras_mobilenet_mixed_precision.ipynb) | &#x2705; PTQ <br/> &#x2705; Mixed-Precision |
| [Nanodet-Plus](../imx500_notebooks/keras/example_keras_nanodet_plus_for_imx500.ipynb) | &#x2705; PTQ |
| [YoloV8-nano](keras/example_keras_yolov8n.ipynb) | &#x2705; PTQ |
| Tutorial | Included Features |
|--------------------------------------------------------------------------------------------|-----------------------------------------------------------------------------------------------------|
| [Basic Post-Training Quantization (PTQ)](keras/example_keras_post-training_quantization.ipynb) | &#x2705; PTQ |
| [MobileNetV2](../imx500_notebooks/keras/example_keras_mobilenetv2_for_imx500.ipynb) | &#x2705; PTQ |
| [Mixed-Precision MobileNetV2](keras/example_keras_mobilenet_mixed_precision.ipynb) | &#x2705; PTQ <br/> &#x2705; Mixed-Precision |
| [Nanodet-Plus](../imx500_notebooks/keras/example_keras_nanodet_plus_for_imx500.ipynb) | &#x2705; PTQ |
| [EfficientDetLite0](../imx500_notebooks/keras/example_keras_effdet_lite0_for_imx500.ipynb) | &#x2705; PTQ <br/> &#x2705; [sony-custom-layers](https://github.com/sony/custom_layers) integration |

</details>
Expand Down
Loading

0 comments on commit fb13d32

Please sign in to comment.