o oiJ@sdddlmZddlmZddlmZddlmZddlm Z ddl m Z m Z dgZ GdddZd S) )nn)tensor) Normalize)DexiNed) EdgeDetector)ResizePostProcessorResizePreProcessorDexiNedBuilderc @s0eZdZdZed dedededefd d Z d S) r zDexiNedBuilder is a class that builds a DexiNed model. .. code-block:: python images = kornia.utils.sample.get_sample_images() model = DexiNedBuilder.build() model.save(images) dexinedT` model_name pretrained image_sizereturncCsp|dkr$ttgdgtdgdgd}t|t|dt}ntd|dt|t ||t dd S) Nr )g ףp= ?gv/?gCl?gp?)meanstd)r zModel z) not found. Please choose from 'DexiNed'.)name) lowerrrr SequentialrSigmoid ValueErrorrrr)r r rnormmodelrX/home/ubuntu/.local/lib/python3.10/site-packages/kornia/models/edge_detection/dexined.pybuild's "zDexiNedBuilder.buildN)r Tr ) __name__ __module__ __qualname____doc__ staticmethodstrboolintrrrrrrr s "N)torchr kornia.corerkornia.enhance.normalizerkornia.filters.dexinedr!kornia.models.edge_detection.baserkornia.models.utilsrr__all__r rrrrs