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.
13 lines
319 B
13 lines
319 B
from ..builder import DETECTORS |
|
from .cascade_rcnn import CascadeRCNN |
|
|
|
|
|
@DETECTORS.register_module() |
|
class HybridTaskCascade(CascadeRCNN): |
|
|
|
def __init__(self, **kwargs): |
|
super(HybridTaskCascade, self).__init__(**kwargs) |
|
|
|
@property |
|
def with_semantic(self): |
|
return self.roi_head.with_semantic
|
|
|