Spaces:
Sleeping
Sleeping
from mmengine.config import read_base | |
from mmdet.models import BatchFixedSizePad | |
from seg.models.data_preprocessor import VideoSegDataPreprocessor | |
with read_base(): | |
from ..._base_.default_runtime import * | |
from ..._base_.datasets.youtube_vis_2021 import * | |
from ..._base_.schedules.schedule_12e import * | |
from ..._base_.datasets.joint_dataset import train_dataloader as training_loader | |
batch_augments = [ | |
dict( | |
type=BatchFixedSizePad, | |
size=(image_size[1], image_size[0]), | |
img_pad_value=0, | |
pad_mask=True, | |
mask_pad_value=0, | |
pad_seg=True, | |
seg_pad_value=255 | |
) | |
] | |
data_preprocessor = dict( | |
type=VideoSegDataPreprocessor, | |
mean=[123.675, 116.28, 103.53], | |
std=[58.395, 57.12, 57.375], | |
bgr_to_rgb=True, | |
pad_size_divisor=32, | |
pad_mask=True, | |
mask_pad_value=0, | |
pad_seg=True, | |
seg_pad_value=255, | |
batch_augments=batch_augments | |
) | |
num_things_classes = 40 | |
num_stuff_classes = 0 | |
num_classes = num_things_classes + num_stuff_classes | |
ov_datasets_name = 'YouTubeVISDataset_2021' | |
default_hooks.update( | |
logger=dict(type=LoggerHook, interval=1), | |
) | |