burn-train 0.21.0-pre.5

Training crate for the Burn framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

//! LPIPS (Learned Perceptual Image Patch Similarity) metric module.
//!
//! LPIPS measures perceptual similarity between images using deep features.
//! Supports VGG16, AlexNet, and SqueezeNet as backbone networks.
//!
//! Reference: "The Unreasonable Effectiveness of Deep Features as a Perceptual Metric"
//! <https://arxiv.org/abs/1801.03924>

mod alexnet;
mod metric;
mod squeezenet;
mod vgg;
mod weights;

pub use metric::{Lpips, LpipsAlex, LpipsConfig, LpipsNet, LpipsSqueeze, LpipsVgg};
pub use weights::{get_backbone_weights_url, get_lpips_weights_url, load_pretrained_weights};

// Re-export feature extractors for advanced use cases
pub use alexnet::AlexFeatureExtractor;
pub use squeezenet::{FireModule, SqueezeFeatureExtractor};
pub use vgg::VggFeatureExtractor;