Skip to content

Instantly share code, notes, and snippets.

@farukcankaya

farukcankaya/detectron2_custom_MaskRCNNConvUpsampleHead.py

Created October 3, 2022 14:52
Show Gist options
  • Save farukcankaya/3dcadb8df44c12ac320d601a5ef6ce9f to your computer and use it in GitHub Desktop.
Save farukcankaya/3dcadb8df44c12ac320d601a5ef6ce9f to your computer and use it in GitHub Desktop.
Download ZIP
Raw
detectron2_custom_MaskRCNNConvUpsampleHead.py
@ROI_MASK_HEAD_REGISTRY.register()
class CustomMaskRCNNConvUpsampleHead(MaskRCNNConvUpsampleHead):
@configurable
def __init__(self, input_shape: ShapeSpec, *, num_classes, conv_dims, conv_norm="", **kwargs):
super().__init__(input_shape, num_classes=num_classes, conv_dims=conv_dims, conv_norm=conv_norm, **kwargs)
self.vis_period = 0
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment