add dry option

Author: VanDavv

cli/blobconverter/__init__.py

@@ -194,7 +194,7 @@ def __download_from_response(resp, fpath: Path):
 
 
 def compile_blob(blob_name, version=None, shaves=None, req_data=None, req_files=None, output_dir=None, url=None,
-                  use_cache=True, compile_params=None, data_type=None, download_ir=False, zoo_type=None):
+                  use_cache=True, compile_params=None, data_type=None, download_ir=False, zoo_type=None, dry=False):
     if shaves is None:
         shaves = __defaults["shaves"]
     if url is None:
@@ -225,6 +225,7 @@ def compile_blob(blob_name, version=None, shaves=None, req_data=None, req_files=
     url_params = {
         'version': version,
         'no_cache': not use_cache,
+        'dry': dry,
     }
     data = {
         "myriad_shaves": str(shaves),
@@ -288,6 +289,9 @@ def compile_blob(blob_name, version=None, shaves=None, req_data=None, req_files=
             print(response.text)
     response.raise_for_status()
 
+    if dry:
+        return response.json()
+
     blob_path.parent.mkdir(parents=True, exist_ok=True)
     if download_ir:
         blob_path = blob_path.with_suffix('.zip')
@@ -489,6 +493,7 @@ def __run_cli__():
     parser.add_argument('--compile-params', help="Additional params to use when compiling a model to MyriadX blob")
     parser.add_argument('--converter-url', dest="url", help="URL to BlobConverter API endpoint used for conversion")
     parser.add_argument('--no-cache', dest="use_cache", action="store_false", help="Omit .cache directory and force new compilation of the blob")
+    parser.add_argument('--dry', dest="dry", action="store_true", help="Instead of compiling the blob, return compilation commands (for manual conversion)")
     parser.add_argument('--zoo-list', action="store_true", help="List all models available in OpenVINO Model Zoo")
     parser.add_argument('--download-ir', action="store_true", help="Downloads OpenVINO IR files used to compile the blob. Result path points to a result ZIP archive")
 
@@ -498,7 +503,7 @@ def __run_cli__():
 
     common_args = {
         arg: getattr(args, arg)
-        for arg in ["shaves", "data_type", "output_dir", "version", "url", "compile_params", "download_ir", "zoo_type"]
+        for arg in ["shaves", "data_type", "output_dir", "version", "url", "compile_params", "download_ir", "zoo_type", "use_cache", "dry"]
     }
     if args.zoo_list:
         return zoo_list()

cli/blobconverter/test.py

@@ -9,7 +9,19 @@
 if not use_cache and Path(blobconverter.__defaults["output_dir"]).exists():
     shutil.rmtree(blobconverter.__defaults["output_dir"])
 
-result = blobconverter.from_zoo(name="mobilenet-ssd", use_cache=use_cache)
+result = blobconverter.from_zoo(name="mobilenet-ssd", use_cache=use_cache, dry=True)
+print(result)
+
+result = blobconverter.from_openvino(
+    xml="../../face-detection-retail-0004.xml",  # get from https://storage.openvinotoolkit.org/repositories/open_model_zoo/2021.2/models_bin/3/face-detection-retail-0004/FP16/face-detection-retail-0004.xml
+    bin="../../face-detection-retail-0004.bin",  # get from https://storage.openvinotoolkit.org/repositories/open_model_zoo/2021.2/models_bin/3/face-detection-retail-0004/FP16/face-detection-retail-0004.bin
+    data_type="FP16",
+    shaves=5,
+    use_cache=use_cache,
+)
+print(result)
+
+result = blobconverter.from_zoo(name="mobilenet-ssd", use_cache=use_cache, dry=True)
 print(result)
 
 result = blobconverter.from_openvino(

cli/setup.py

@@ -4,7 +4,7 @@
 
 setup(
     name='blobconverter',
-    version='1.2.4',
+    version='1.2.5',
     description='The tool that allows you to convert neural networks to MyriadX blob',
     long_description=io.open("README.md", encoding="utf-8").read(),
     long_description_content_type="text/markdown",