Spaces:
Runtime error
Runtime error
File size: 2,095 Bytes
b34d1d6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
# dataset settings
# do not use this config for training, it is only used to create embedding.
from mmengine import read_base
from mmengine.dataset import DefaultSampler, RepeatDataset
from seg.datasets.concat_dataset import ConcatOVDataset
from seg.datasets.samplers.batch_sampler import VideoSegAspectRatioBatchSampler
with read_base():
from .coco_panoptic_lsj import train_dataloader as _coco_vid_train_loader
from .ade_panoptic_ov import train_dataloader as _ade_train_loader
from .youtube_vis_2019 import train_dataloader as _yt19_train_loader
from .youtube_vis_2021 import train_dataloader as _yt21_train_loader
from .vipseg import train_dataloader as _vipseg_train_loader
from .cityscapes_panoptic import train_dataloader as _city_train_loader
from .coco_panoptic_lsj import val_dataloader, val_evaluator, test_dataloader, test_evaluator
from .youtube_vis_2019 import image_size
train_dataloader = dict(
batch_size=1,
num_workers=2,
persistent_workers=True,
sampler=dict(type=DefaultSampler, shuffle=True),
batch_sampler=dict(type=VideoSegAspectRatioBatchSampler),
dataset=dict(
type=ConcatOVDataset,
datasets=[
dict(
type=RepeatDataset,
dataset=_coco_vid_train_loader.dataset,
times=1,
),
dict(
type=RepeatDataset,
dataset=_ade_train_loader.dataset,
times=1,
),
dict(
type=RepeatDataset,
dataset=_yt19_train_loader.dataset,
times=1,
),
dict(
type=RepeatDataset,
dataset=_yt21_train_loader.dataset,
times=1,
),
dict(
type=RepeatDataset,
dataset=_vipseg_train_loader.dataset,
times=1,
),
dict(
type=RepeatDataset,
dataset=_city_train_loader.dataset,
times=1,
),
],
)
)
|