Spaces:
Sleeping
Sleeping
File size: 1,015 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 |
from mmengine.config import read_base
from mmdet.models import BatchFixedSizePad
from seg.models.data_preprocessor import VideoSegDataPreprocessor
from seg.models.utils import NO_OBJ
with read_base():
from ..._base_.default_runtime import *
from ..._base_.datasets.davis 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=NO_OBJ
)
]
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=NO_OBJ,
batch_augments=batch_augments
)
num_things_classes = 80
num_stuff_classes = 0
num_classes = num_things_classes + num_stuff_classes
ov_datasets_name = 'CocoOVDataset'
|