GitHubリポジトリOpenVINO™その他」ブランチからトレーニング拡張機能をインストールするにはどうすればよいですか?

GitHub* リポジトリ その他のブランチからトレーニング拡張機能をインストールできませんOpenVINO™。

前提 条件

• Ubuntu* 18.04 / 20.04
• Python* 3.6+
• OpenVINO™ - モデルのエクスポートと実行用
• CUD ツールキット 10.2 - GPU でのトレーニング向け

インストール手順:

1. Linux* OS 向け OpenVINO ツールキット 2021.4.2 をダウンロード。
2. Linux* 向けのインテル® ディストリビューションの OpenVINO™ ツールキットのインストールと構成 に従ってください。
3. クローン OpenVINO™トレーニング拡張機能リポジトリ:

   git clone https://github.com/openvinotoolkit/training_extensions.git

   export OTE_DIR=`pwd`/training_extensions

4. クローン Model Zoo リポジトリーを開いてデモを実行します。

   git clone https://github.com/openvinotoolkit/open_model_zoo --branch 2021.4.2

   export OMZ_DIR=`pwd`/open_model_zoo

5. ディレクトリーを object_detection に変更します。

   cd /training_extensions/models/object_detection

6. 仮想環境の作成:

   ./init_venv.sh

7. 仮想環境をアクティブ化します。

   source venv/bin/activate

8. 次の runtime.txt ファイルで onnxoptimizer と onnx のバージョンを変更します。

   /training_extensions/external/mmdetection/requirements/runtime.txt

   onnx==1.10.1

   onnxoptimizer==0.2.6

9. インストール 仮想環境内でのMMDetection:

   cd /training_extensions/external/mmdetection/

   pip install -r requirements/build.txt

   pip install "git+ https://github.com/open-mmlab/cocoapi.git#subdirectory=pycocotools"

   pip install -v -e .

10. 水平テキスト検出ガイド の手順 2 に進みます。

    cd /training_extensions/models/object_detection

    export MODEL_TEMPLATE=`realpath ./model_templates/horizontal-text-detection/horizontal-text-detection-0001/template.yaml`

    export WORK_DIR=/tmp/my-$(basename $(dirname $MODEL_TEMPLATE))

    export SNAPSHOT=snapshot.pth

    python ../../tools/instantiate_template.py ${MODEL_TEMPLATE} ${WORK_DIR}

11. 手順 3 に進みます。事前トレーニング済みモデルを試す

    cd ${WORK_DIR}

    python export.py \

    --load-weights ${SNAPSHOT} \

    --save-model-to export

    python ${OMZ_DIR}/demos/object_detection_demo/python/object_detection_demo.py \

    -m export/model.xml \

    -at ssd \

    -i /dev/video0

トラブルシューティングの手順:

ModuleNotFoundError: 'mmcv._ext' という名前のモジュールが発生していない場合は、CUDA をバージョン 10.2 にダウングレードします。

RuntimeError: OpenVINO™ Model Optimizer が見つからないか、正しく構成されていない場合は、「ステップ 4: モデル・オプティマイザーを構成する」の手順を実行します。

エラーが続く場合は、/mmdetection/tools/export.py ファイル内の mo.py のフルパスを入力してください (154、171、179 および 185 行目)。

mo.py のフルパス:

/opt/intel/openvino_2021.4.752/deployment_tools/model_optimizer/mo.py