Skip to content

Commit

Permalink
Merge branch 'main' into 22-separate-our-annotation-file-into-train-v…
Browse files Browse the repository at this point in the history
…al-test
  • Loading branch information
Grutschus committed Nov 24, 2023
2 parents 02943a5 + 60063c2 commit 8bd708d
Showing 1 changed file with 11 additions and 15 deletions.
26 changes: 11 additions & 15 deletions configs/datasets/ds_uniformsample_existencelabel.py
Original file line number Diff line number Diff line change
@@ -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

0 comments on commit 8bd708d

Please sign in to comment.