o oi@s<ddlmZmZddlmZddlmZGdddeZdS))ModuleTensor)KORNIA_CHECK_SHAPE)DexiNedcsjeZdZdZdfdd ZdeddfddZd edefd d Zd edefd dZ d edefddZ Z S) EdgeDetectoranDetect edges in a given image using a CNN. By default, it uses the method described in :cite:`xsoria2020dexined`. Return: A tensor of shape :math:`(B,1,H,W)`. Example: >>> img = torch.rand(1, 3, 320, 320) >>> detect = EdgeDetector() >>> out = detect(img) >>> out.shape torch.Size([1, 1, 320, 320]) returnNcsttdd|_dS)NT) pretrained)super__init__rmodel)self __class__Q/home/ubuntu/.local/lib/python3.10/site-packages/kornia/contrib/edge_detection.pyr (s zEdgeDetector.__init__ path_filecCs|j|dSN)r load_from_file)r rrrrload,szEdgeDetector.loadimagecC|Srr)r rrrr preprocess/zEdgeDetector.preprocessdatacCrrr)r rrrr postprocess2rzEdgeDetector.postprocesscCs,t|gd||}||}||S)N)B3HW)rrr r)r rimgoutrrrforward5s   zEdgeDetector.forward)rN) __name__ __module__ __qualname____doc__r strrrrrr! __classcell__rrr rrsrN) kornia.corerrkornia.core.checkrkornia.filters.dexinedrrrrrrs