2?jm- d Z ddlZddlmZ ddlmZmZmZmZm Z ddl
Z
ddl Z ddl
mZ ddlmZmZmZ ddlmZmZ dd lmZ G d
d e ZdS ) ag Read/Write images using OpenCV.
Backend Library: `OpenCV `_
This plugin wraps OpenCV (also known as ``cv2``), a popular image processing
library. Currently, it exposes OpenCVs image reading capability (no video or GIF
support yet); however, this may be added in future releases.
Methods
-------
.. note::
Check the respective function for a list of supported kwargs and their
documentation.
.. autosummary::
:toctree:
OpenCVPlugin.read
OpenCVPlugin.iter
OpenCVPlugin.write
OpenCVPlugin.properties
OpenCVPlugin.metadata
Pixel Formats (Colorspaces)
---------------------------
OpenCV is known to process images in BGR; however, most of the python ecosystem
(in particular matplotlib and other pydata libraries) use the RGB. As such,
images are converted to RGB, RGBA, or grayscale (where applicable) by default.
N)Path)AnyDictListOptionalUnion )Request) URI_BYTESInitializationErrorIOMode)ImagePropertiesPluginV3) ArrayLikec
` e Zd Zdeddf fd Zddej ddedeee f dede
j fd Z dej fdeee f dede
j fd
Z
dd eeee f d
edee dee fdZddej fdedeee f dedefdZ ddededee ef fdZ xZS ) OpenCVPluginrequestreturnNc t | | | _ |j t
u rd| _ n |j | _ |j j }|t j k r1t j
| j st d| j d |t j k r1t j | j st d| j d d S d S )NzzOpenCV can't read ``.zOpenCV can't write to `)super__init__get_local_filename file_handle _uri_typer filenameraw_urimodeio_moder
readcv2haveImageReaderr writehaveImageWriter)selfr r __class__s GC:\Projects\Historical\venv\Lib\site-packages\imageio/plugins/opencv.pyr zOpenCVPlugin.__init__/ s
!!!"5577 ) ) %DMM #ODM|# 6; s':4;K'L'L %&MDM&M&M&MNN N
V\
!
!#*=d>N*O*O
!%&Q
&Q&Q&QRR R "
!
!
! )index
colorspaceflagsr) r* r+ c
|$t j | j | }|dk rdnd}|du r!t j | j | \ }}d}n"t j | j |d| \ }}d}|du rt d| d | j d
|d j d k rd }|pd } n$|d j d
dk rd}|pd} nd}|pd} t |t r|
nJ||
k rd
n/|
} t t d| d|
!t j
fd|D }nt j
| }|r|n|d S )a Read an image from the ImageResource.
Parameters
----------
index : int, Ellipsis
If int, read the index-th image from the ImageResource. If ``...``,
read all images from the ImageResource and stack them along a new,
prepended, batch dimension. If None (default), use ``index=0`` if
the image contains exactly one image and ``index=...`` otherwise.
colorspace : str, int
The colorspace to convert into after loading and before returning
the image. If None (default) keep grayscale images as is, convert
images with an alpha channel to ``RGBA`` and all other images to
``RGB``. If int, interpret ``colorspace`` as one of OpenCVs
`conversion flags
`_
and use it for conversion. If str, convert the image into the given
colorspace. Possible string values are: ``"RGB"``, ``"BGR"``,
``"RGBA"``, ``"BGRA"``, ``"GRAY"``, ``"HSV"``, or ``"LAB"``.
flags : int
The OpenCV flag(s) to pass to the reader. Refer to the `OpenCV docs
`_
for details.
Returns
-------
ndimage : np.ndarray
The decoded image as a numpy array.
N r .)r+ TFzCould not read index `z` from `r r GRAY BGRARGBABGRRGBCOLOR_2c : g | ]}t j | S )r! cvtColor).0x cvt_spaces r'
z%OpenCVPlugin.read.. s% DDD1CLI 6 6DDDr( )r! imcountr imreadmulti
ValueErrorr ndimshape
isinstanceintuppergetattrnpstack) r% r) r* r+ n_imagesretvalimgis_batch
in_colorspaceout_colorspacer< s @r' r zOpenCVPlugin.read> s L ={4#3U;;H!Q AACE C<< /$*:%HHHKFCHH /$*:E1ERRRKFC H U?? Ve V VT] V V VWW W q6;! "M ' 16NN
V\"
"
" "M ' 16NN !M ' 05N j# & & P"II
n 2 2 4 4
4
4 II + 1 1 3 3N %Nm%N%Nn%N%NOOI (DDDDDDDEECC(3--C *ssCF*r( c # K t t j | j D ] }| ||| V dS )aU Yield images from the ImageResource.
Parameters
----------
colorspace : str, int
The colorspace to convert into after loading and before returning
the image. If None (default) keep grayscale images as is, convert
images with an alpha channel to ``RGBA`` and all other images to
``RGB``. If int, interpret ``colorspace`` as one of OpenCVs
`conversion flags
`_
and use it for conversion. If str, convert the image into the given
colorspace. Possible string values are: ``"RGB"``, ``"BGR"``,
``"RGBA"``, ``"BGRA"``, ``"GRAY"``, ``"HSV"``, or ``"LAB"``.
flags : int
The OpenCV flag(s) to pass to the reader. Refer to the `OpenCV docs
`_
for details.
Yields
------
ndimage : np.ndarray
The decoded image as a numpy array.
r) r+ r* N)ranger! r> r r )r% r* r+ idxs r' iterzOpenCVPlugin.iter s[ < T%56677 K KC))#Uz)JJ J J J J K Kr( FndimagerL paramsc t |t rt j |d }n
|s|d }|d j dk rd}n|d j d }|dk r
d |D }n |dk r
d |D }n d
|D }t
j | j || }|d u rt d | j
j t k r&t | j S d
S )a$ Save an ndimage in the ImageResource.
Parameters
----------
ndimage : ArrayLike, List[ArrayLike]
The image data that will be written to the file. It is either a
single image, a batch of images, or a list of images.
is_batch : bool
If True, the provided ndimage is a batch of images. If False (default), the
provided ndimage is a single image. If the provided ndimage is a list of images,
this parameter has no effect.
params : List[int]
A list of parameters that will be passed to OpenCVs imwrite or
imwritemulti functions. Possible values are documented in the
`OpenCV documentation
`_.
Returns
-------
encoded_image : bytes, None
If the ImageResource is ``""`` the call to write returns the
encoded image as a bytes string. Otherwise it returns None.
r )axis)N.r r- r/ c g | ]}|S r8 r8 r: r; s r' r= z&OpenCVPlugin.write.. s ...1...r( r0 c L g | ]!}t j |t j "S r8 )r! r9 COLOR_RGBA2BGRArY s r' r= z&OpenCVPlugin.write.. s' QQQA3<3+>??QQQr( c L g | ]!}t j |t j "S r8 )r! r9
COLOR_RGB2BGRrY s r' r= z&OpenCVPlugin.write.. s' OOO!3<3+<==OOOr( FzOpenCV failed to write.N)rC listrG rH rA rB r! imwritemultir IOErrorr r r r
read_bytes)r% rT rL rU
n_channels ndimage_cv2rJ s r' r# zOpenCVPlugin.write s > gt $ $ )hwQ///GG ) i(G 1:?a JJ )"-J ??..g...KK
1__QQQQQKKOOwOOOK!$"2KHH U?? 344 4 < !Y . .())4466 6 / .r( c v |!t j | j | }|dk }n(|t u r t j | j | }d}nd}|r9| d|| }t |g|j R |j |d S | ||| }t |j |j d S ) aV Standardized image metadata.
Parameters
----------
index : int, Ellipsis
If int, get the properties of the index-th image in the
ImageResource. If ``...``, get the properties of the image stack
that contains all images. If None (default), use ``index=0`` if the
image contains exactly one image and ``index=...`` otherwise.
colorspace : str, int
The colorspace to convert into after loading and before returning
the image. If None (default) keep grayscale images as is, convert
images with an alpha channel to ``RGBA`` and all other images to
``RGB``. If int, interpret ``colorspace`` as one of OpenCVs
`conversion flags
`_
and use it for conversion. If str, convert the image into the given
colorspace. Possible string values are: ``"RGB"``, ``"BGR"``,
``"RGBA"``, ``"BGRA"``, ``"GRAY"``, ``"HSV"``, or ``"LAB"``.
flags : int
The OpenCV flag(s) to pass to the reader. Refer to the `OpenCV docs
`_
for details.
Returns
-------
props : ImageProperties
A dataclass filled with standardized image metadata.
Notes
-----
Reading properties with OpenCV involves decoding pixel data, because
OpenCV doesn't provide a direct way to access metadata.
Nr- TFr rP )rB dtyperI rL )rB re rL )r! r> r Ellipsisr r rB re )r% r) r* r+ rI rL rK s r'
propertieszOpenCVPlugin.properties s T ={4#3U;;H !|HH
h
{4#3U;;HHH H ))!5Z)HHC" ,#),,i!
iie5ZiHH SYci%PPPPr( Texclude_appliedc R t j dt t S )aj Format-specific metadata.
.. warning::
OpenCV does not support reading metadata. When called, this function
will raise a ``NotImplementedError``.
Parameters
----------
index : int
This parameter has no effect.
exclude_applied : bool
This parameter has no effect.
z)OpenCV does not support reading metadata.)warningswarn UserWarningdict)r% r) rh s r' metadatazOpenCVPlugin.metadata& s $
A;OOOvv
r( )FN)NT)__name__
__module__ __qualname__r
r r! IMREAD_COLORrD r strrG ndarrayr rS r r boolr bytesr# r rg r r rn
__classcell__)r& s @r' r r . s
S
SD
S
S
S
S
S
S$ &*% K+ K+ K+ K+ #s(O K+
K+
K+ K+ K+ K+^ '+% K K#s(O K K
K K K KH 87 87y$y/1287 87 S 87
% 87 87 87 87x &*% >Q >Q>Q #s(O>Q >Q
>Q >Q >Q >QB :> 26
c3h r( r )__doc__rj pathlibr typingr r r r r r! numpyrG corer
core.requestr r r
core.v3_plugin_apir r r r r8 r( r' r s @ 3 3 3 3 3 3 3 3 3 3 3 3 3 3
A A A A A A A A A A : : : : : : : : K K K K K8 K K K K Kr(