Function tch_utils::gabor::gabor_filter
source · pub fn gabor_filter(
size: usize,
theta: f64,
sigma: f64,
lambda: f64,
psi: f64,
gamma: f64,
device: Device
) -> TensorExpand description
Generates a Gabor filter.
Arguments
- size: usize - The size of the filter
- theta: f64 - The orientation of the filter in degrees
- sigma: f64 - The standard deviation of the Gaussian envelope
- lambda: f64 - The wavelength of the sinusoidal factor
- psi: f64 - The phase offset of the sinusoidal factor
- gamma: f64 - The spatial aspect ratio
- device: Device - The device to store the tensor on
Returns
- [size, size] float tensor - The Gabor filter