From 9fe28706d0e5f055ca85d62483eeef040e39ae6c Mon Sep 17 00:00:00 2001 From: Maxim Vafin Date: Wed, 2 Jun 2021 18:58:25 +0300 Subject: [PATCH] Fix YOLACT documentation (#5990) * Fix YOLACT documentation * Update Convert_YOLACT.md --- .../convert_model/pytorch_specific/Convert_YOLACT.md | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/docs/MO_DG/prepare_model/convert_model/pytorch_specific/Convert_YOLACT.md b/docs/MO_DG/prepare_model/convert_model/pytorch_specific/Convert_YOLACT.md index ed072ac64f4..9fb7e1ca9e9 100644 --- a/docs/MO_DG/prepare_model/convert_model/pytorch_specific/Convert_YOLACT.md +++ b/docs/MO_DG/prepare_model/convert_model/pytorch_specific/Convert_YOLACT.md @@ -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: