Module tch_utils::image

source ·
Expand description

Adds a trait to convert between tch::Tensor and image::DynamicImage.

Traits

  • ImageTensorExt
    Trait to convert between tch::Tensor and image::DynamicImage.