Fix e2e (#2179)

* Fix iseg e2e (#2173) * Fix det e2e (#2176) * Fix det e2e * Fix mypy
openvinotoolkit · May 23, 2023 · 4b2453d · 4b2453d
1 parent 70ae88c
commit 4b2453d
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/otx/algorithms/detection/configs/detection/cspdarknet_yolox/data_pipeline.py b/otx/algorithms/detection/configs/detection/cspdarknet_yolox/data_pipeline.py
@@ -48,7 +48,7 @@
         img_scale=(416, 416),
         flip=False,
         transforms=[
-            dict(type="Resize", keep_ratio=False),
+            dict(type="Resize", keep_ratio=True),
             dict(type="RandomFlip"),
             dict(type="Pad", size=(416, 416), pad_val=114.0),
             dict(type="Normalize", **__img_norm_cfg),

diff --git a/otx/api/configuration/helper/convert.py b/otx/api/configuration/helper/convert.py
@@ -8,7 +8,7 @@
 
 
 from enum import Enum
-from typing import Type, TypeVar
+from typing import Any, Type, TypeVar
 
 import yaml
 from omegaconf import DictConfig, OmegaConf
@@ -128,6 +128,7 @@ def convert(
         config_id = config_dict.get("id", None)
         config_dict["id"] = str(config_id) if config_id is not None else None
 
+    result: Any = None
     if target == str:
         result = yaml.dump(config_dict)
     elif target == dict:

diff --git a/tests/e2e/cli/detection/test_detection.py b/tests/e2e/cli/detection/test_detection.py
@@ -46,7 +46,7 @@
     "--val-data-roots": "tests/assets/car_tree_bug",
     "--test-data-roots": "tests/assets/car_tree_bug",
     "--input": "tests/assets/car_tree_bug/images/train",
-    "train_params": ["params", "--learning_parameters.num_iters", "6", "--learning_parameters.batch_size", "4"],
+    "train_params": ["params", "--learning_parameters.num_iters", "15", "--learning_parameters.batch_size", "4"],
 }
 
 # Class-Incremental learning w/ 'vehicle', 'person', 'non-vehicle' classes

diff --git a/tests/e2e/cli/instance_segmentation/test_instance_segmentation.py b/tests/e2e/cli/instance_segmentation/test_instance_segmentation.py
@@ -51,7 +51,7 @@
     "--val-data-roots": "tests/assets/car_tree_bug",
     "--test-data-roots": "tests/assets/car_tree_bug",
     "--input": "tests/assets/car_tree_bug/images/train",
-    "train_params": ["params", "--learning_parameters.num_iters", "4", "--learning_parameters.batch_size", "2"],
+    "train_params": ["params", "--learning_parameters.num_iters", "5", "--learning_parameters.batch_size", "2"],
 }
 
 # Training params for resume, num_iters*2