Fix YOLACT documentation (#5990)

* Fix YOLACT documentation

* Update Convert_YOLACT.md
This commit is contained in:
Maxim Vafin authored and GitHub committed 2021-06-02 18:58:25 +03:00
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: