gausplat_trainer/metric/
mod.rs

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
28
//! Metrics for evaluation.

pub mod mae;
pub mod mdssim;
pub mod mse;
pub mod mssim;
pub mod psnr;

pub use burn::tensor::{backend::Backend, Tensor};
pub use mae::*;
pub use mdssim::*;
pub use mse::*;
pub use mssim::*;
pub use psnr::*;

/// Functionality for evaluating metrics.
pub trait Metric<B: Backend> {
    /// Evaluate the value against the target.
    ///
    /// ## Returns
    ///
    /// The metric value with shape `[1]`.
    fn evaluate<const D: usize>(
        &self,
        value: Tensor<B, D>,
        target: Tensor<B, D>,
    ) -> Tensor<B, 1>;
}