Custom targets for albumentations

albumentations_targets.py

# import base class first
# or you'll get such error:
# TypeError: super(type, obj): obj must be an instance or subtype of type
from albumentations import DualTransform

# new targets function
def new_tf_targets(self):
    return {
        'image': self.apply, # do not rename this one
        'image2': self.apply, # new
        'target': self.apply, # new
        'mask': self.apply_to_mask,
        'bboxes': self.apply_to_bboxes
    }

# redefine targets function
# don't forget to wrap it in property
DualTransform.targets = property(new_tf_targets)

# now import all dependent classes
from albumentations import (
    ...
)