Attend the Women in ML Symposium on December 7 Register now

Module: tfm.vision.maskrcnn

Stay organized with collections Save and categorize content based on your preferences.

MaskRCNN task definition.

Classes

class MaskRCNNTask: A single-replica view of training procedure.

Functions

zero_out_disallowed_class_ids(...): Zero out IDs of classes not in allowed_class_ids.