MGEloss

class aixd.mlmodel.architecture.losses.MGEloss(*args: Any, **kwargs: Any)[source]

Bases: Module

Mean gradient error. Uses a sobel filter to compute the gradient of predicted and estimated images, and then applies a mean square difference of the pixel-wise gradients.

Methods