NAFNet/basicsr/models/losses/__init__.py

12 lines
480 B
Python

# ------------------------------------------------------------------------
# Copyright (c) 2022 megvii-model. All Rights Reserved.
# ------------------------------------------------------------------------
# Modified from BasicSR (https://github.com/xinntao/BasicSR)
# Copyright 2018-2020 BasicSR Authors
# ------------------------------------------------------------------------
from .losses import (L1Loss, MSELoss, PSNRLoss)
__all__ = [
'L1Loss', 'MSELoss', 'PSNRLoss',
]