From d3d4be13c76147435477139a838942f8bf087908 Mon Sep 17 00:00:00 2001 From: Eugene Khvedchenya Date: Tue, 28 May 2024 16:18:51 +0300 Subject: [PATCH] Move limitations to the top --- notebooks/YoloNAS_R_Export_to_ONNX.ipynb | 33 +++++++++--------------- 1 file changed, 12 insertions(+), 21 deletions(-) diff --git a/notebooks/YoloNAS_R_Export_to_ONNX.ipynb b/notebooks/YoloNAS_R_Export_to_ONNX.ipynb index 1794775d3..f7e8b142e 100644 --- a/notebooks/YoloNAS_R_Export_to_ONNX.ipynb +++ b/notebooks/YoloNAS_R_Export_to_ONNX.ipynb @@ -48,9 +48,20 @@ "- Customising input image shape and batch size\n", "- Customising NMS parameters and number of detections per image\n", "- Customising output format (flat or batched)\n", - "- Exported ONNX model can be used with TensorRT" + "- Exported ONNX model can be used with TensorRT\n", + "\n", + "### Limitations\n", + "\n", + "- Dynamic batch size / input image shape is not supported yet. You can only export a model with a fixed batch size and input image shape.\n", + "- TensorRT of version 8.6 or higher is required.\n", + "- Quantization to FP16 requires CUDA / MPS device available." ] }, + { + "metadata": {}, + "cell_type": "markdown", + "source": "" + }, { "cell_type": "code", "execution_count": 1, @@ -1125,26 +1136,6 @@ "show_predictions_from_flat_format(image, result)" ] }, - { - "cell_type": "markdown", - "metadata": { - "collapsed": false, - "id": "KZK3oB3EG7ba", - "jupyter": { - "outputs_hidden": false - }, - "pycharm": { - "name": "#%% md\n" - } - }, - "source": [ - "### Limitations\n", - "\n", - "* Dynamic batch size / input image shape is not supported yet. You can only export a model with a fixed batch size and input image shape.\n", - "* TensorRT of version 8.6 or higher is required.\n", - "* Quantization to FP16 requires CUDA / MPS device available." - ] - }, { "cell_type": "markdown", "metadata": {