File size: 1,322 Bytes
2568013 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
from .loss import Loss
from .loss_depth import LossDepth, LossDepthCfgWrapper
from .loss_lpips import LossLpips, LossLpipsCfgWrapper
from .loss_mse import LossMse, LossMseCfgWrapper
from .loss_opacity import LossOpacity, LossOpacityCfgWrapper
from .loss_depth_gt import LossDepthGT, LossDepthGTCfgWrapper
from .loss_lod import LossLOD, LossLODCfgWrapper
from .loss_depth_consis import LossDepthConsis, LossDepthConsisCfgWrapper
from .loss_normal_consis import LossNormalConsis, LossNormalConsisCfgWrapper
from .loss_chamfer_distance import LossChamferDistance, LossChamferDistanceCfgWrapper
LOSSES = {
LossDepthCfgWrapper: LossDepth,
LossLpipsCfgWrapper: LossLpips,
LossMseCfgWrapper: LossMse,
LossOpacityCfgWrapper: LossOpacity,
LossDepthGTCfgWrapper: LossDepthGT,
LossLODCfgWrapper: LossLOD,
LossDepthConsisCfgWrapper: LossDepthConsis,
LossNormalConsisCfgWrapper: LossNormalConsis,
LossChamferDistanceCfgWrapper: LossChamferDistance,
}
LossCfgWrapper = LossDepthCfgWrapper | LossLpipsCfgWrapper | LossMseCfgWrapper | LossOpacityCfgWrapper | LossDepthGTCfgWrapper | LossLODCfgWrapper | LossDepthConsisCfgWrapper | LossNormalConsisCfgWrapper | LossChamferDistanceCfgWrapper
def get_losses(cfgs: list[LossCfgWrapper]) -> list[Loss]:
return [LOSSES[type(cfg)](cfg) for cfg in cfgs]
|