[78161] Show fastseg models in OMZ table (#25)

Author: vgoluben

automation/bom/image_BOM.txt

@@ -38,7 +38,6 @@ migrations/versions/1e1c4fc30e7c_add_accuracy_import.py
 migrations/versions/2133a18b601e_load_text_dataset_support.py
 migrations/versions/241ced8ef5aa_create_project_report_pipeline.py
 migrations/versions/260ffba4b041_create_inference_report_pipeline.py
-migrations/versions/275bd4d6cd35_instant_assets_sharing.py
 migrations/versions/2a070ca81f20_tensor_distance_report_query_api.py
 migrations/versions/39bd11960375_edge_node_script_in_deployment.py
 migrations/versions/424ed0531227_add_annotate_dataset_job.py
@@ -67,6 +66,7 @@ migrations/versions/8e7b97ccae4d_add_ubuntu20_support.py
 migrations/versions/9af85a4ff36c_add_operating_system_to_targets.py
 migrations/versions/9dd49b6b0552_early_visualization.py
 migrations/versions/9eae56bc5c8e_add_wider_and_openimages_datasets.py
+migrations/versions/9ede8e012999_enable_fastseg_models.py
 migrations/versions/a291f54f85ec_add_new_yolo_task_methods.py
 migrations/versions/a4bffbc50716_add_visualization_config.py
 migrations/versions/a511acc6ec24_add_opset_in_model_details.py
@@ -78,6 +78,7 @@ migrations/versions/b6ba1e22cc13_add_reshape_model_pipeline.py
 migrations/versions/b9bfeb130aff_dataset_creation_to_pipeline.py
 migrations/versions/bc1a7d48829c_generate_accuracy_reports_for_accuracy_.py
 migrations/versions/bec2e0d3859d_add_accuracy_report.py
+migrations/versions/c2f12b313b48_instant_assets_sharing.py
 migrations/versions/c2ff2481cc2e_add_inference_test_image.py
 migrations/versions/c7e09f4ff6e9_fix_enums.py
 migrations/versions/d1d5c84cafbe_remote_accuracy.py

migrations/versions/9ede8e012999_enable_fastseg_models.py

@@ -0,0 +1,47 @@
+"""Enable fastseg models to appear in the OMZ table
+
+Revision ID: 9ede8e012999
+Revises: c2f12b313b48
+Create Date: 2022-02-11 15:23:34.127963
+
+"""
+
+"""
+ OpenVINO DL Workbench
+ Migration: Enable fastseg models to appear in the OMZ table
+
+ Copyright (c) 2022 Intel Corporation
+
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+      http://www.apache.org/licenses/LICENSE-2.0
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+"""
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '9ede8e012999'
+down_revision = 'c2f12b313b48'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+
+    op.execute(
+        'UPDATE omz_topologies SET advanced_configuration = \'{"preprocessing": [{"type": "auto_resize"}], "postprocessing": [{"type": "encode_segmentation_mask", "apply_to": "annotation"}, {"type": "resize_segmentation_mask", "apply_to": "prediction"}], "metric": [{"type": "mean_iou", "use_argmax": false, "presenter": "print_scalar", "reference": 0.7267}], "annotationConversion": {}, "adapterConfiguration": {"adapter": {"type": "segmentation", "make_argmax": true}}}\' WHERE name = \'fastseg-large\';')
+    op.execute(
+        'UPDATE omz_topologies SET advanced_configuration = \'{"preprocessing": [{"type": "auto_resize"}], "postprocessing": [{"type": "encode_segmentation_mask", "apply_to": "annotation"}, {"type": "resize_segmentation_mask", "apply_to": "prediction"}], "metric": [{"type": "mean_iou", "use_argmax": false, "presenter": "print_scalar", "reference": 0.6715}], "annotationConversion": {}, "adapterConfiguration": {"adapter": {"type": "segmentation", "make_argmax": true}}}\' WHERE name = \'fastseg-small\';')
+
+
+def downgrade():
+    raise NotImplementedError(f'Downgrade is not implemented for the {revision} migration')
+