Computes and stores the average and current value

A registry to map strings to classes or functions. Registered object could be built from registry. Meanwhile, registered functions could be ca

A named tuple describing a ResNet block.

Mobile Inverted Residual Bottleneck Block. Args: block_args (namedtuple): BlockArgs, defined in utils.py. global_params (namedtupl

Component model mask

DFL facial mask

Extended mask Based on components mask. Attempts to extend the eyebrow points up the forehead

Basic face hull mask

Block Decoder for readability, straight from the official TensorFlow repository.

nn.Module warpper for focal loss

PolyLoss: A Polynomial Expansion Perspective of Classification Loss Functions

2D Convolutions like TensorFlow, for a dynamic image size. The padding is operated in forward function by calculating dynamically.

2D Convolutions like TensorFlow's 'SAME' mode, with the given input image size. The padding mudule is calculated in construction function, then

EfficientNet model. Most easily loaded with the .from_name or .from_pretrained methods. Args: blocks_args (list[namedtuple]): A lis

Parent class for masks the output mask will be <mask_type>.mask channels: 1, 3 or 4: 1 - Returns a single channel

Xception optimized for the ImageNet dataset, as specified in https://arxiv.org/pdf/1610.02357.pdf