Edit this page

Import Original Model Recommendations

Use the following recommendations to make the process of importing original models easier:

  1. Check model format

    Сheck whether the model format is supported in the DL Workbench, for example, supported formats include TensorFlow*, ONNX*, OpenVINO™ Intermediate Representation (IR), and others.

  2. Convert model to supported format

    Optional. Convert the model if its format is not directly supported.

  3. Import model in DL Workbench

    Obtain a model in Intermediate Representation (IR) format to start working with the DL Workbench.

1. Check Model Format

The import process depends on the framework of your model. Examples of the supported formats: TensorFlow* (Frozen graph, Checkpoint, Metagraph, SavedModel, Keras H5), ONNX*, OpenVINO™ IR.

  • If the model is already in the IR format, no further steps are required. Import the model files and proceed to create a project.

  • Optional. If the model is in one of the supported formats, go to Step 3 and import it in the DL Workbench to obtain a model in IR format.

  • If the model is not in one of the supported formats, proceed to the next step to convert it to a supported format and then to IR.

2. Convert Model to Supported Format

To convert the model to one of the input formats supported in the DL Workbench, find a specific format you need to use for your model:

  • If the model is from PyTorch framework, it is recommended to convert it to the ONNX format.

  • If the model is trained in another framework, look for a similar topology supported in the Open Model Zoo (OMZ).

    1. Open model.yml file in the OMZ repository, where you can find the information on:

      • which input format should be used ​for the model of your framework

      • how to convert the model to this format

      • how to configure accuracy of the model

    2. In the model.yml file find the Model Optimizer arguments required to convert the model:

      ../_images/model_optimizer_args.png

    3. Pay attention to the file extension of the model (for example, .savedmodel, .onnx, .pb, and others). You need to convert your model to the specified format. In the example above, you can see that yolov4 needs to be converted to the TensorFlow .savedmodel format.

  • If there is no similar model for your use case, it is recommended to use either ONNX or TensorFlow formats. Refer to the Model Optimizer Developer Guide to learn more general information on importing models from different frameworks.

3. Import Model in DL Workbench

As a result of importing your model in the DL Workbench, you will obtain the model in IR format. For that, go to the DL Workbench and click Create Project and Import Model.

../_images/cp_yolo.png ../_images/import_start_yolo.png

Select the framework and provide the folder with the model in one of the supported format. Next, you need to fill out the import form in the DL Workbench. This information concerns the model training and depends on the framework, for example, the way the input images were fed to the Computer Vision model during training:

  • color space (BGR, RGB, Grayscale)

  • applied transformations (mean subtraction, scale factor multiplication)

../_images/import_params.png

After importing your model in the DL Workbench, you can download the IR model and work with OpenVINO components or proceed to accelerate performance, visualize model architecture and outputs, benchmark inferencing performance, measure accuracy, and explore other DL Workbench capabilities.

../_images/model_download.png

See Also

Prev Next