Wie installiere ich OpenVINO™ Trainingserweiterung aus dem GitHub-Repository-Zweig "Misc"?

OpenVINO™ Training Extension kann nicht aus dem GitHub*-Repository misc branch installiert werden.

Voraussetzungen

• Ubuntu* 18.04 / 20.04
• Python* 3.6+
• OpenVINO™ - zum Exportieren und Ausführen von Modellen
• CUDA Toolkit 10.2 – für das Training mit GPU

Installationsschritte:

1. Laden Sie OpenVINO Toolkit 2021.4.2 für Linux* OS herunter.
2. Folgen Sie der Anleitung Intel® Distribution des OpenVINO™ Toolkit installieren und konfigurieren für Linux*.
3. Klon OpenVINO™ Repository für Schulungserweiterungen:

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

   export OTE_DIR=`pwd`/training_extensions

4. Klon Öffnen Sie das Model Zoo-Repository, um Demos auszuführen:

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

   export OMZ_DIR=`pwd`/open_model_zoo

5. Ändern Sie das Verzeichnis in object_detection:

   cd /training_extensions/models/object_detection

6. Erstellen einer virtuellen Umgebung:

   ./init_venv.sh

7. Aktivieren Sie die virtuelle Umgebung:

   source venv/bin/activate

8. Ändern Sie die Versionen onnxoptimizer und onnx in der folgenden runtime.txt-Datei:

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

   onnx==1.10.1

   onnxoptimizer==0.2.6

9. Installieren MMDetektion in der virtuellen Umgebung:

   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. Fahren Sie mit Schritt 2 im Leitfaden zur horizontalen Texterkennung fort.

    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. Fahren Sie mit Schritt 3 fort. Probieren Sie ein vortrainiertes Modell aus

    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

Schritte zur Fehlerbehebung:

Wenn ModuleNotFoundError: Es ist kein Modul mit dem Namen "mmcv._ext" aufgetreten, führen Sie ein Downgrade von CUDA auf Version 10.2 durch.

Wenn RuntimeError: OpenVINO™ Model Optimizer nicht gefunden wird oder nicht ordnungsgemäß konfiguriert ist, führen Sie die Schritte in Schritt 4: Konfigurieren des Model Optimizer aus.

Wenn der Fehler weiterhin besteht, geben Sie den vollständigen PATH für mo.py in der Datei /mmdetection/tools/export.py an (Zeile 154, 171, 179 und 185).

VOLLSTÄNDIGER PFAD für mo.py:

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