|
from detectron2.config import LazyCall as L
|
|
from detectron2.layers import ShapeSpec
|
|
from detectron2.modeling import PanopticFPN
|
|
from detectron2.modeling.meta_arch.semantic_seg import SemSegFPNHead
|
|
|
|
from .mask_rcnn_fpn import model
|
|
|
|
model._target_ = PanopticFPN
|
|
model.sem_seg_head = L(SemSegFPNHead)(
|
|
input_shape={
|
|
f: L(ShapeSpec)(stride=s, channels="${....backbone.out_channels}")
|
|
for f, s in zip(["p2", "p3", "p4", "p5"], [4, 8, 16, 32])
|
|
},
|
|
ignore_value=255,
|
|
num_classes=54,
|
|
conv_dims=128,
|
|
common_stride=4,
|
|
loss_weight=0.5,
|
|
norm="GN",
|
|
)
|
|
|
