<filename>icevision/models/torchvision/mask_rcnn/fastai/callbacks.py
__all__ = ["MaskRCNNCallback"]

from icevision.models.torchvision.fastai.callbacks import *
from icevision.models.torchvision.mask_rcnn.prediction import *


class MaskRCNNCallback(RCNNCallback):
    def convert_raw_predictions(self, raw_preds):
        return convert_raw_predictions(
            raw_preds=raw_preds, detection_threshold=0.0, mask_threshold=0.0
        )

<filename>icevision/models/torchvision/mask_rcnn/fastai/callbacks.py
__all__ = ["MaskRCNNCallback"]

from icevision.models.torchvision.fastai.callbacks import *
from icevision.models.torchvision.mask_rcnn.prediction import *


class MaskRCNNCallback(RCNNCallback):
    def convert_raw_predictions(self, raw_preds):
        return convert_raw_predictions(
            raw_preds=raw_preds, detection_threshold=0.0, mask_threshold=0.0
        )
