Skip to content

Commit b1cc38b

Browse files
eugene123twkprokofi
authored andcommitted
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 8cc1961 commit b1cc38b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

+1-1
Original file line numberDiff line numberDiff line change
@@ -276,7 +276,7 @@ def adapt_tile_config(tile_config: TileConfig, dataset: Dataset, task: OTXTaskTy
276276
task (Task): task type of the model
277277
"""
278278
if (train_dataset := dataset.subsets().get("train") or dataset.subsets().get("TRAINING")) is not None:
279-
stat = compute_robust_dataset_statistics(train_dataset, task=task)
279+
stat = compute_robust_dataset_statistics(train_dataset)
280280
max_num_objects = round(stat["annotation"]["num_per_image"]["max"])
281281
avg_size = stat["annotation"]["size_of_shape"]["avg"]
282282
min_size = stat["annotation"]["size_of_shape"]["robust_min"]

0 commit comments

Comments
 (0)