Skip to content

Commit aa31dca

Browse files
authored
Include Geti arrow dataset subset names (#3962)
* restrited number of output masks by tiling * add geti subset name * update num of max pred
1 parent b1ec8e7 commit aa31dca

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/otx/core/data/utils/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -250,7 +250,7 @@ def adapt_tile_config(tile_config: TileConfig, dataset: Dataset) -> None:
250250
tile_config (TileConfig): tiling parameters of the model
251251
dataset (Dataset): Datumaro dataset including all subsets
252252
"""
253-
if (train_dataset := dataset.subsets().get("train")) is not None:
253+
if (train_dataset := dataset.subsets().get("train") or dataset.subsets().get("TRAINING")) is not None:
254254
stat = compute_robust_dataset_statistics(train_dataset)
255255
max_num_objects = round(stat["annotation"]["num_per_image"]["max"])
256256
avg_size = stat["annotation"]["size_of_shape"]["avg"]

0 commit comments

Comments
 (0)