HarborYuan's picture
add omg code
b34d1d6
raw
history blame
No virus
1.06 kB
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_2019 import *
from ..._base_.schedules.schedule_12e import *
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_2019'
default_hooks.update(
logger=dict(type=LoggerHook, interval=1),
)