Roman Kazantsev
826f345daf
* [TF FE] Support Switch and Merge to fuse into If operation It introduces support of TF1 control flow with Switch and Merge nodes. Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com> * Add script for test model generation * Fix code-style Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com> * Fix build issue Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com> * Fix build issue with types Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com> * Apply code-review feedback: optimizations in utils Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com> * Fix build issue Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com> * Apply code-review remarks and cover more cases Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com> * Remove commented code Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com> * Remove unused vars Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com> * Update MO unit-tests wit Switch-Merge case Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com> * Fix build issue: remove unused variable Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com> --------- Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>
Installation
Installing from PyPi
- Create a virtual environment and activate it, e.g.:
virtualenv -p /usr/bin/python3.7 .env3
source .env3/bin/activate
- Install openvino-dev package, it contains model conversion API:
pip install openvino-dev
This will download all requirements and will install MO in your current virtual environment. If you need only particular frameworks you can specify them manually as optional dependencies in square brackets. E.g. the command below will install dependencies to support ONNX* and TensorFlow2* models:
pip install openvino-dev[onnx,tensorflow2]
To enable support of MxNet* models run:
pip install openvino-dev[mxnet]
To enable support of all frameworks:
pip install openvino-dev[all]
By default, if no frameworks are specified, dependencies to support ONNX* and TensorFlow2* are installed.
Converting models
Setup development environment
If you want to contribute to model conversion API you will need to deploy developer environment. You can do that by following the steps below:
- Create virtual environment and activate it, e.g.:
virtualenv -p /usr/bin/python3.7 .env3
source .env3/bin/activate
- Clone the OpenVINO™ repository and change dir to model-optimizer
git clone https://github.com/openvinotoolkit/openvino
cd openvino/tools/mo/
- Install openvino-mo package for development:
pip install -e .
or run setup.py develop, result will be the same:
python setup.py develop
This will download all requirements and deploy model conversion API for development in your virtual environment: specifically will create *.egg-link into the current directory in your site-packages. As previously noted, you can also manually specify to support only selected frameworks :
pip install -e ".[onnx,tensorflow2]"
How to run unit-tests
- Run tests with:
python -m unittest discover -p "*_test.py" [-s PATH_TO_DIR]
How to capture unit-tests coverage
- Run tests with:
coverage run -m unittest discover -p "*_test.py" [-s PATH_TO_DIR]
- Build html report:
coverage html
How to run code linting
- Run the following command:
pylint openvino/tools/mo/ mo.py