Skip to content

Commit

Permalink
Minor fixes in bug-report and README (#180)
Browse files Browse the repository at this point in the history
* Update README.md

* Minor fixes in bug-report and README.md
  • Loading branch information
zhiqwang committed Sep 30, 2021
1 parent 00f6e13 commit 06022fd
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 5 deletions.
2 changes: 1 addition & 1 deletion .github/ISSUE_TEMPLATE/bug-report.yml
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ body:
description: |
Please provide a clear and concise description of what the bug is.
If relevant, add a minimal example so that we can reproduce the error by running the code. It is very important for he snippet to be as succinct (minimal) as possible, so please take time to trim down any irrelevant code to help us debug efficiently. We are going to copy-paste your code and we expect to get the same result as you did: avoid any external data, and include the relevant imports, etc. For example:
If relevant, add a minimal example so that we can reproduce the error by running the code. It is very important for the snippet to be as succinct (minimal) as possible, so please take time to trim down any irrelevant code to help us debug efficiently. We are going to copy-paste your code and we expect to get the same result as you did: avoid any external data, and include the relevant imports, etc. For example:
```python
# All necessary imports at the beginning
Expand Down
7 changes: 3 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -112,12 +112,9 @@ The following is the interface for loading the checkpoint weights trained with `
```python
from yolort.models import YOLOv5

# Model
yolov5 = YOLOv5()

# 'yolov5s.pt' is downloaded from https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt
ckpt_path_from_ultralytics = 'yolov5s.pt'
model = yolov5.load_from_yolov5(ckpt_path_from_ultralytics, score_thresh=0.25)
model = YOLOv5.load_from_yolov5(ckpt_path_from_ultralytics, score_thresh=0.25)

model.eval()
img_path = 'test/assets/bus.jpg'
Expand All @@ -132,6 +129,8 @@ We provide a [notebook](notebooks/inference-pytorch-export-libtorch.ipynb) to de

### Inference on `ONNXRuntime` backend

TBD

## 🎨 Model Graph Visualization

Now, `yolort` can draw the model graph directly, checkout our [model-graph-visualization](notebooks/model-graph-visualization.ipynb) notebook to see how to use and visualize the model graph.
Expand Down

0 comments on commit 06022fd

Please sign in to comment.