From 60063c20debb2ae5bd8b2876aa215a17304fe9ec Mon Sep 17 00:00:00 2001 From: Till Grutschus Date: Wed, 22 Nov 2023 16:25:04 +0100 Subject: [PATCH] hotfix: dataset config example --- .../ds_uniformsample_existencelabel.py | 26 ++++++++----------- 1 file changed, 11 insertions(+), 15 deletions(-) diff --git a/configs/datasets/ds_uniformsample_existencelabel.py b/configs/datasets/ds_uniformsample_existencelabel.py index 14e2438..1cf0100 100644 --- a/configs/datasets/ds_uniformsample_existencelabel.py +++ b/configs/datasets/ds_uniformsample_existencelabel.py @@ -1,22 +1,18 @@ -from datasets.transforms.label_strategy import HQFD_LABEL_DESCRIPTION - custom_imports = dict(imports="datasets", allow_failed_imports=False) type = "HighQualityFallDataset" sampling_strategy = dict(type="UniformSampling", clip_len=10) -label_strategy = dict(type="ExistenceLabel", label_description=HQFD_LABEL_DESCRIPTION) +label_strategy = dict( + type="ExistenceLabel", + label_description=dict( + names=["fall", "lying", "other"], + start_timestamp_names=["fall_start", "lying_start"], + end_timestamp_names=["fall_end", "lying_end"], + visible_names=["fall_visible", "lying_visible"], + other_class=2, + ), +) ann_file = "tests/test_data/test_annotation.csv" -pipeline = [ - dict(type="DecordInit"), - dict(type="ClipVideo"), - dict(type="SampleFrames", clip_len=16, frame_interval=4, num_clips=1), - dict(type="DecordDecode"), - dict(type="Resize", scale=(-1, 224)), - dict(type="RandomResizedCrop"), - dict(type="Resize", scale=(224, 224), keep_ratio=False), - dict(type="Flip", flip_ratio=0.5), - dict(type="FormatShape", input_format="NCTHW"), - dict(type="PackActionInputs"), -] # type: ignore +pipeline = [] # type: ignore multiclass = True num_classes = 3 test_mode = True