Update colab export notebook and README.

Author: Christian Mills

README.md

@@ -5,14 +5,21 @@ This tutorial provides a step-by-step guide on training YOLOX object detection m
 
 
 
-## Training Code
+## Training
 
 | Jupyter Notebook                                             | Google Colab                                                 |
 | ------------------------------------------------------------ | ------------------------------------------------------------ |
 | [GitHub Repository](https://github.com/cj-mills/pytorch-yolox-object-detection-tutorial-code/blob/main/notebooks/pytorch-yolox-object-detector-training.ipynb) | [Open In Colab](https://colab.research.google.com/github/cj-mills/pytorch-yolox-object-detection-tutorial-code/blob/main/notebooks/pytorch-yolox-object-detector-training-colab.ipynb) |
 
 
 
+## ONNX Export
+
+| Jupyter Notebook                                             | Google Colab                                                 |
+| ------------------------------------------------------------ | ------------------------------------------------------------ |
+| [GitHub Repository](https://github.com/cj-mills/pytorch-yolox-object-detection-tutorial-code/blob/main/notebooks/pytorch-yolox-object-detector-onnx-export.ipynb) | [Open In Colab](https://colab.research.google.com/github/cj-mills/pytorch-yolox-object-detection-tutorial-code/blob/main/notebooks/pytorch-yolox-object-detector-onnx-export-colab.ipynb) |
+
+
 
 ## License
 

notebooks/pytorch-yolox-object-detector-onnx-export-colab.ipynb

@@ -15,18 +15,18 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# %%capture\n",
-    "# # Install PyTorch with CUDA\n",
-    "# !pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118\n",
+    "%%capture\n",
+    "# Install PyTorch with CUDA\n",
+    "!pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118\n",
     "\n",
-    "# # Install additional dependencies\n",
-    "# !pip install pandas pillow\n",
+    "# Install additional dependencies\n",
+    "!pip install pandas pillow\n",
     "\n",
-    "# # Install ONNX packages\n",
-    "# !pip install onnx onnxruntime onnx-simplifier\n",
+    "# Install ONNX packages\n",
+    "!pip install onnx onnxruntime onnx-simplifier\n",
     "\n",
-    "# # Install utility packages\n",
-    "# !pip install cjm_psl_utils cjm_pil_utils cjm_yolox_pytorch"
+    "# Install utility packages\n",
+    "!pip install cjm_psl_utils cjm_pil_utils cjm_yolox_pytorch"
    ]
   },
   {

notebooks/pytorch-yolox-object-detector-training.ipynb

@@ -2845,6 +2845,34 @@
      },
      "metadata": {},
      "output_type": "display_data"
+    },
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "1c25da16e74a448090294f499b19225c",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "Eval:   0%|          | 0/99 [00:00<?, ?it/s]"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "3c835b13774e4bd8a17c45f4fb320d44",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "Train:   0%|          | 0/895 [00:00<?, ?it/s]"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
     }
    ],
    "source": [