Skip to content

Commit 43eb838

Browse files
authored
Remove CenterCrop from Classification test pipeline and editing missing docs link (#2375)
* Fix missing link for docs and removing centercrop for classification data pipeline * Revert the test threshold
1 parent 85bfe9f commit 43eb838

File tree

3 files changed

+5
-7
lines changed

3 files changed

+5
-7
lines changed

docs/source/guide/explanation/algorithms/segmentation/semantic_segmentation.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ Dataset Format
4242

4343
For the dataset handling inside OpenVINO™ Training Extensions, we use `Dataset Management Framework (Datumaro) <https://github.com/openvinotoolkit/datumaro>`_.
4444

45-
At this end we support `Common Semantic Segmentation <https://openvinotoolkit.github.io/datumaro/docs/formats/common_semantic_segmentation/>`_ data format.
45+
At this end we support `Common Semantic Segmentation <https://github.com/openvinotoolkit/datumaro/blob/develop/docs/source/docs/data-formats/formats/common_semantic_segmentation.md>`_ data format.
4646
If you organized supported dataset format, starting training will be very simple. We just need to pass a path to the root folder and desired model template to start training:
4747

4848
.. code-block::
@@ -278,4 +278,4 @@ It can be launched only with supervised (incremental) training type.
278278
.. Incremental Learning
279279
.. ********************
280280
281-
.. To be added soon
281+
.. To be added soon

src/otx/algorithms/classification/configs/base/data/data_pipeline.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
__resize_target_size = 224
2121

2222
__train_pipeline = [
23-
dict(type="RandomResizedCrop", size=224, efficientnet_style=True),
23+
dict(type="RandomResizedCrop", size=__resize_target_size, efficientnet_style=True),
2424
dict(type="RandomFlip", flip_prob=0.5, direction="horizontal"),
2525
dict(type="Normalize", **__img_norm_cfg),
2626
dict(type="ImageToTensor", keys=["img"]),
@@ -29,8 +29,7 @@
2929
]
3030

3131
__test_pipeline = [
32-
dict(type="Resize", size=(256, -1)),
33-
dict(type="CenterCrop", crop_size=224),
32+
dict(type="Resize", size=__resize_target_size),
3433
dict(type="Normalize", **__img_norm_cfg),
3534
dict(type="ImageToTensor", keys=["img"]),
3635
dict(type="Collect", keys=["img"]),

tests/e2e/cli/classification/test_classification.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -180,8 +180,7 @@ def test_otx_explain_openvino(self, template, tmp_dir_path):
180180
@pytest.mark.parametrize("half_precision", [True, False])
181181
def test_otx_eval_openvino(self, template, tmp_dir_path, half_precision):
182182
tmp_dir_path = tmp_dir_path / "multi_class_cls"
183-
# FIXME [Jaeguk] Revert threshold to 0.2 when model api supports resize and centercrop.
184-
otx_eval_openvino_testing(template, tmp_dir_path, otx_dir, args, threshold=0.5, half_precision=half_precision)
183+
otx_eval_openvino_testing(template, tmp_dir_path, otx_dir, args, threshold=0.2, half_precision=half_precision)
185184

186185
@e2e_pytest_component
187186
@pytest.mark.skipif(TT_STABILITY_TESTS, reason="This is TT_STABILITY_TESTS")

0 commit comments

Comments
 (0)