|
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", |
|
) |
|
|
