Update README and Rename jupyter notebook

zhiqwang · Nov 21, 2020 · eb18d27 · eb18d27
1 parent a0f5e2c
commit eb18d27
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -27,9 +27,9 @@ python -m utils.updated_checkpoint [--checkpoint_path ./yolov5s.pt]
                                    [--updated_checkpoint_path ./checkpoints/yolov5/yolov5s.pt]
 ```
 
-### ✨ PyTorch interface
+### ✨ `PyTorch` Interface
 
-The `detect.py` reads a image source and does inference.
+The `detect.py` reads a image source and does inference, you can also check for the more details in [inference-pytorch-export-libtorch](notebooks/inference-pytorch-export-libtorch.ipynb) notebook.
 
 ```bash
 python -m detect [--model_cfg ./models/yolov5s.yaml]
@@ -43,10 +43,16 @@ python -m detect [--model_cfg ./models/yolov5s.yaml]
                  [--gpu]  # GPU switch, Set False as default
 ```
 
-### 🚀 LibTorch interface
+### 🚀 `LibTorch` Interface
 
 Here providing an [example](./deployment) of getting `LibTorch` inferencing to work. Also you can check the [CI](.github/workflows/stable.yml) for more details.
 
+### ✏️ Model Visualization
+
+Now, `yolov5-rt-stack` can draw the model graph directly, check for more details in [visualize-jit-models](notebooks/visualize-jit-models.ipynb) notebook.
+
+<a href="notebooks/assets/yolov5.detail.svg"><img src="notebooks/assets/yolov5.detail.svg" alt="YOLO model visualize" width="500"/></a>
+
 ## 🎓 Acknowledgement
 
 - The implementation of `yolov5` borrow the code from [ultralytics](https://github.com/ultralytics/yolov5).

diff --git a/...books/inference-on-pytorch-libtorch.ipynb → ...s/inference-pytorch-export-libtorch.ipynb b/...books/inference-on-pytorch-libtorch.ipynb → ...s/inference-pytorch-export-libtorch.ipynb