OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
15 lines
450 B
15 lines
450 B
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
|
|
|