Skip to content

Commit

Permalink
Fix PR comments
Browse files Browse the repository at this point in the history
  • Loading branch information
Idan-BenAmi committed Nov 12, 2023
1 parent d399870 commit c60fc1e
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tutorials/notebooks/example_keras_yolov8n.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -133,9 +133,10 @@
"cell_type": "markdown",
"source": [
"### Generate Yolo8-nano Keras model\n",
"In the following steps, we proceed to create a Keras model based on [Ultralytics](https://github.com/ultralytics/ultralytics). We then initialize the model weights to match those of the original model.\n",
"In the following steps, we proceed to create a Keras model based on [Ultralytics](https://github.com/ultralytics/ultralytics). This implementation includes a slightly modified version of yolov8\n",
"detection-head (mainly the box decoding part) that was optimized for model quantization. We then initialize the model weights to match those of the original model.\n",
"\n",
"To this base model, we integrate the post-processing components, which include box decoding layers following by tensorflow [tf.image.combined_non_max_suppression](https://www.tensorflow.org/api_docs/python/tf/image/combined_non_max_suppression) layer.\n",
"To this base model, we integrate the post-processing components, which include tensorflow [tf.image.combined_non_max_suppression](https://www.tensorflow.org/api_docs/python/tf/image/combined_non_max_suppression) layer.\n",
"For further insights into the model's implementation details, please refer to the tutorials [\"resources\" folder](https://github.com/sony/model_optimization/tree/main/tutorials/resources). "
],
"metadata": {
Expand Down

0 comments on commit c60fc1e

Please sign in to comment.