Add anchor / target assignment visualization notebook (#116)

* Cleanup training samples visualization * Add anchor assignment visualization notebooks * Update anchor target assignment notebooks * Remove training samples notebooks * Update notebooks in docs * Revert changes in image_utils.py
zhiqwang · Jul 1, 2021 · 82d6afb · 82d6afb
1 parent dbc1c72
commit 82d6afb
Show file tree

Hide file tree

Showing 9 changed files with 478 additions and 209 deletions.
diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -12,14 +12,15 @@ and with modules refactoring to make it available in deployment backends such as
    installation
 
 .. toctree::
-   :maxdepth: 2
+   :maxdepth: 3
    :caption: Examples:
 
-   notebooks/load_model_as_ultralytics
-   notebooks/visualize_jit_models
+   notebooks/how_to_align_with_ultralytics_yolov5
+   notebooks/model_graph_visualization
+   notebooks/anchor_label_assignment_visualization
 
 .. toctree::
-   :maxdepth: 2
+   :maxdepth: 3
    :caption: Deployment:
 
    notebooks/inference_pytorch_export_libtorch

diff --git a/docs/source/notebooks/.gitignore b/docs/source/notebooks/.gitignore
@@ -0,0 +1,2 @@
+# notebook exceptions
+!*.ipynb
diff --git a/docs/source/notebooks/anchor_label_assignment_visualization.ipynb b/docs/source/notebooks/anchor_label_assignment_visualization.ipynb
@@ -0,0 +1 @@
+../../../notebooks/anchor-label-assignment-visualization.ipynb
diff --git a/docs/source/notebooks/how_to_align_with_ultralytics_yolov5.ipynb b/docs/source/notebooks/how_to_align_with_ultralytics_yolov5.ipynb
@@ -0,0 +1 @@
+../../../notebooks/how-to-align-with-ultralytics-yolov5.ipynb
diff --git a/docs/source/notebooks/load_model_as_ultralytics.ipynb b/docs/source/notebooks/load_model_as_ultralytics.ipynb
diff --git a/docs/source/notebooks/model_graph_visualization.ipynb b/docs/source/notebooks/model_graph_visualization.ipynb
@@ -0,0 +1 @@
+../../../notebooks/model-graph-visualization.ipynb
diff --git a/docs/source/notebooks/visualize_jit_models.ipynb b/docs/source/notebooks/visualize_jit_models.ipynb
diff --git a/notebooks/anchor-label-assignment-visualization.ipynb b/notebooks/anchor-label-assignment-visualization.ipynb
diff --git a/notebooks/training-samples-visualization.ipynb b/notebooks/training-samples-visualization.ipynb