Greenaug

dataphy.visionpack.adapters.greenaug

Classes

GreenAugAdapter(texture_bank: str = './textures')

Bases: Module

Source code in src/dataphy/visionpack/adapters/greenaug.py

def __init__(self, texture_bank: str = "./textures"):
    super().__init__()
    self.texture_bank = texture_bank

Attributes

texture_bank = texture_bank instance-attribute

Functions

forward(batch)

Source code in src/dataphy/visionpack/adapters/greenaug.py

def forward(self, batch):
    # TODO: integrate real green-screen compositor
    return batch