Spaces:
Sleeping
Sleeping
| """Models module""" | |
| from .encoder import MobileNetV3Encoder, get_encoder | |
| from .decoder import UNetLiteDecoder, get_decoder | |
| from .network import ForgeryLocalizationNetwork, get_model | |
| from .losses import DiceLoss, CombinedLoss, DatasetAwareLoss, get_loss_function | |
| __all__ = [ | |
| 'MobileNetV3Encoder', | |
| 'get_encoder', | |
| 'UNetLiteDecoder', | |
| 'get_decoder', | |
| 'ForgeryLocalizationNetwork', | |
| 'get_model', | |
| 'DiceLoss', | |
| 'CombinedLoss', | |
| 'DatasetAwareLoss', | |
| 'get_loss_function' | |
| ] | |