Skip to content

Commit eb13c97

Browse files
authored
allow to configure out the mask_head of the HTC algorithm (open-mmlab#5389)
The HTC algorithm has an interleaved execution of bbox_heads and mask_heads. For Object-Detection we only need the bbox_head. The design of the HTC algorithms allows to remove the mask_head while the bbox_head keeps working. Co-authored-by: Maximilian Otten <[email protected]>
1 parent 4c62867 commit eb13c97

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mmdet/models/roi_heads/htc_roi_head.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def __init__(self,
2525
**kwargs):
2626
super(HybridTaskCascadeRoIHead,
2727
self).__init__(num_stages, stage_loss_weights, **kwargs)
28-
assert self.with_bbox and self.with_mask
28+
assert self.with_bbox
2929
assert not self.with_shared_head # shared head is not supported
3030

3131
if semantic_head is not None:

0 commit comments

Comments
 (0)