o toi5@s<ddlmZddlZddlZddlmZGdddeZdS)) annotationsN)OpRunc@seZdZd d ddZdS) ImageDecoderRGBencoded np.ndarrayreturntuple[np.ndarray]c Cszddl}Wnty}ztd|d}~ww|jt|}|dkr;t|dddddddf}|fS|dkrGt|}|fS|dkr_| d}t|}tj |dd }|fSt d |d ) NrzYPillow must be installed to use the reference implementation of the ImageDecoder operatorBGRr GrayscaleL)axisz pixel_format=z is not supported.) PIL.Image ImportErrorImageopenioBytesIOtobytesnparrayconvert expand_dims ValueError)selfr pixel_formatPILeimgdecodedr"W/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_image_decoder.py_runs. "    zImageDecoder._runN)r)rrrr )__name__ __module__ __qualname__r$r"r"r"r#rsr) __future__rrnumpyronnx.reference.op_runrrr"r"r"r#s