fast-sobel-tfjs
Version:
GPU-accelerated Sobel edge detection for TensorFlow.js - 5-10x faster than CPU implementations
25 lines (24 loc) • 738 B
TypeScript
import { KernelSize } from './types';
/**
* Kernel definitions for Sobel filters of different sizes
*/
interface KernelDefinitions {
x: Record<KernelSize, number[][]>;
y: Record<KernelSize, number[][]>;
}
/**
* Pre-defined Sobel kernels for different sizes
*/
export declare const KERNELS: KernelDefinitions;
/**
* Validates whether a kernel size is supported
* @param size The kernel size to validate
* @returns True if the kernel size is supported, false otherwise
*/
export declare function isValidKernelSize(size: number): size is KernelSize;
/**
* Gets the available kernel sizes as an array
* @returns Array of supported kernel sizes
*/
export declare function getAvailableKernelSizes(): KernelSize[];
export {};