File size: 310 Bytes
61c2d32
 
 
 
 
 
 
1
2
3
4
5
6
7
8
# Copyright (c) Facebook, Inc. and its affiliates.
from .config import add_pointrend_config
from .mask_head import PointRendMaskHead, ImplicitPointRendMaskHead
from .semantic_seg import PointRendSemSegHead
from .color_augmentation import ColorAugSSDTransform

from . import roi_heads as _  # only registration