From bc49dbd76c910dfdc22ad91152244d24ae31d9aa Mon Sep 17 00:00:00 2001 From: Eugene Liu Date: Wed, 18 Sep 2024 15:17:11 +0100 Subject: [PATCH 1/3] restrited number of output masks by tiling --- src/otx/core/config/data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/otx/core/config/data.py b/src/otx/core/config/data.py index 5af3016e0e8..cf42094cd32 100644 --- a/src/otx/core/config/data.py +++ b/src/otx/core/config/data.py @@ -76,7 +76,7 @@ class TileConfig: tile_size: tuple[int, int] = (400, 400) overlap: float = 0.2 iou_threshold: float = 0.45 - max_num_instances: int = 1500 + max_num_instances: int = 500 object_tile_ratio: float = 0.03 sampling_ratio: float = 1.0 with_full_img: bool = False From 1d278653b8b09222eafb4d92b46ab4e43cc2de18 Mon Sep 17 00:00:00 2001 From: Eugene Liu Date: Thu, 19 Sep 2024 09:29:19 +0100 Subject: [PATCH 2/3] add geti subset name --- src/otx/core/data/utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/otx/core/data/utils/utils.py b/src/otx/core/data/utils/utils.py index fc0c84dbf58..7bb96fd3588 100644 --- a/src/otx/core/data/utils/utils.py +++ b/src/otx/core/data/utils/utils.py @@ -250,7 +250,7 @@ def adapt_tile_config(tile_config: TileConfig, dataset: Dataset) -> None: tile_config (TileConfig): tiling parameters of the model dataset (Dataset): Datumaro dataset including all subsets """ - if (train_dataset := dataset.subsets().get("train")) is not None: + if (train_dataset := dataset.subsets().get("train") or dataset.subsets().get("TRAINING")) is not None: stat = compute_robust_dataset_statistics(train_dataset) max_num_objects = round(stat["annotation"]["num_per_image"]["max"]) avg_size = stat["annotation"]["size_of_shape"]["avg"] From 69f70c10223c0849410a8b0666bb3db6dd45ab12 Mon Sep 17 00:00:00 2001 From: Eugene Liu Date: Thu, 19 Sep 2024 10:41:47 +0100 Subject: [PATCH 3/3] update num of max pred --- src/otx/core/config/data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/otx/core/config/data.py b/src/otx/core/config/data.py index cf42094cd32..5af3016e0e8 100644 --- a/src/otx/core/config/data.py +++ b/src/otx/core/config/data.py @@ -76,7 +76,7 @@ class TileConfig: tile_size: tuple[int, int] = (400, 400) overlap: float = 0.2 iou_threshold: float = 0.45 - max_num_instances: int = 500 + max_num_instances: int = 1500 object_tile_ratio: float = 0.03 sampling_ratio: float = 1.0 with_full_img: bool = False