mirror of https://github.com/hero-y/BHRL
16 lines
450 B
Python
16 lines
450 B
Python
from ..builder import DETECTORS
|
|
from .cascade_rcnn import CascadeRCNN
|
|
|
|
|
|
@DETECTORS.register_module()
|
|
class HybridTaskCascade(CascadeRCNN):
|
|
"""Implementation of `HTC <https://arxiv.org/abs/1901.07518>`_"""
|
|
|
|
def __init__(self, **kwargs):
|
|
super(HybridTaskCascade, self).__init__(**kwargs)
|
|
|
|
@property
|
|
def with_semantic(self):
|
|
"""bool: whether the detector has a semantic head"""
|
|
return self.roi_head.with_semantic
|