o 0j @shddlZddlmZmZddlmZddlmZedr ddl Z ej ddd edGd d d Z dS) N)class_requires_depsis_dep_available) benchmark) ImageReaderzopencv-contrib-pythonT)nameZis_read_operationcs2eZdZdZd fdd ZddZddZZS) ReadImagezLoad image from the file.BGRcs:t||_tjtjtjd|j}td|d|_dS)z Initialize the instance. Args: format (str, optional): Target color format to convert the image to. Choices are 'BGR', 'RGB', and 'GRAY'. Default: 'BGR'. )r RGBZGRAYZopencv)backendflagsN)super__init__formatcv2Z IMREAD_COLORZIMREAD_GRAYSCALEr _img_reader)selfrr  __class__m/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/common/reader/image_reader.pyrs zReadImage.__init__csfdd|DS)applycsg|]}|qSr)read).0imgrrr 1sz&ReadImage.__call__..r)rZimgsrrr__call__/szReadImage.__call__cCst|tjr|jdkrt|tj}|St|trA|j |}|dur*t d||jdkr?|j dkr8t dt|tj}|St dt|jd)Nr zImage read Error: rzArray is not 3-dimensional.zReadImage only supports the following types: 1. str, indicating a image file path or a directory containing image files. 2. numpy.ndarray. However, got type: .) isinstancenpZndarrayrrZcvtColorZ COLOR_BGR2RGBstrrr Exceptionndim RuntimeError TypeErrortype__name__)rrZblobrrrr3s$      zReadImage.read)r )r( __module__ __qualname____doc__rrr __classcell__rrrrr s r ) numpyr!Z utils.depsrrZutils.benchmarkrZutils.iorrZtimeit_with_optionsr rrrrs