Struct StyleTransfer 

pub struct StyleTransfer {
    pub content_model: Model,
    pub style_model: Model,
    pub iterations: usize,
    pub content_weight: f32,
    pub style_weight: f32,
    pub learning_rate: f32,
}

Style transfer engine.

Fields

content_model: Model

style_model: Model

iterations: usize

content_weight: f32

style_weight: f32

learning_rate: f32

Implementations

impl StyleTransfer

pub fn new(content_model: Model, style_model: Model) -> Self

pub fn from_preset(preset: StylePreset) -> Self

Create a style transfer engine from a preset.

pub fn gram_matrix(features: &Tensor) -> Tensor

Compute the Gram matrix: G = F^T * F, where F has shape (C, N). If the input is flattened or 1-D, reshape to (sqrt, sqrt) approximately.

pub fn content_loss(generated: &Tensor, target: &Tensor) -> f32

Content loss: mean squared error between generated and target features.

pub fn style_loss(generated_gram: &Tensor, target_gram: &Tensor) -> f32

Style loss: MSE between Gram matrices of generated and target features.

pub fn total_loss( &self, gen_content_features: &Tensor, target_content_features: &Tensor, gen_style_features: &Tensor, target_style_features: &Tensor, ) -> f32

Total loss combining content and style.

pub fn transfer(&self, content: &Tensor, style: &Tensor) -> Tensor

Run iterative style transfer optimization. content and style should have the same shape. Returns a generated tensor of the same shape.