Method __init__

src/model/completion_model.py:141–145

(self)

class CompletionNet(BaseModel):
    def __init__(self):
        super().__init__()

        self.videoCN = VideoCN()
        self.combCN = CombCN()

    def forward(self, imgs, masks, guidances=None):
        masked_imgs = imgs * masks