Fix YOLACT documentation (#5990)
* Fix YOLACT documentation * Update Convert_YOLACT.md
This commit is contained in:
1 parent
ac4dd6cda7
commit
9fe28706d0
1 file changed
+6
-3
@@ -136,7 +136,9 @@ git clone https://github.com/dbolya/yolact
|
||||
git checkout 57b8f2d95e62e2e649b382f516ab41f949b57239
|
||||
```
|
||||
|
||||
**Step 2**. Download a pre-trained model, for example `yolact_base_54_800000.pth`.
|
||||
3. Set up the environment as described in `README.md`.
|
||||
|
||||
**Step 2**. Download a pre-trained model from the list attached in the `Evaluation` section of `README.md` document, for example `yolact_base_54_800000.pth`.
|
||||
|
||||
**Step 3**. Export the model to ONNX* format.
|
||||
|
||||
@@ -152,10 +154,11 @@ python3 eval.py \
|
||||
--trained_model=/path/to/yolact_base_54_800000.pth \
|
||||
--score_threshold=0.3 \
|
||||
--top_k=10 \
|
||||
--image=/path/to/image.jpg
|
||||
--image=/path/to/image.jpg \
|
||||
--cuda=False
|
||||
```
|
||||
|
||||
3. You should get `yolact.onnx` file.
|
||||
3. The script may fail, but you should get `yolact.onnx` file.
|
||||
|
||||
**Step 4**. Convert the model to the IR:
|
||||
|
||||
|
||||
Reference in new issue
Block a user