Spaces:
Sleeping
Sleeping
File size: 849 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 |
from mmengine import read_base
from seg.models.detectors import Mask2formerVideoMinVIS
with read_base():
from .datasets.mose import *
from .models.m2_convl_300q import *
model.update(
data_preprocessor=data_preprocessor,
type=Mask2formerVideoMinVIS,
clip_size=5,
clip_size_small=3,
whole_clip_thr=0,
small_clip_thr=15,
overlap=0,
panoptic_head=dict(
ov_classifier_name=f'{ov_model_name}_{ov_datasets_name}',
num_things_classes=num_things_classes,
num_stuff_classes=num_stuff_classes,
),
panoptic_fusion_head=dict(
num_things_classes=num_things_classes,
num_stuff_classes=num_stuff_classes,
),
test_cfg=dict(
panoptic_on=False,
semantic_on=False,
instance_on=False,
proposal_on=True,
num_proposals=25,
),
)
|