^j5UdZddlmZddlZddlZddlZddlmZm Z m Z ddl m Z m Z ddlmZddlmZddlmZmZmZdd lmZ ddlZd Zdd lmZgd ZiZded<iZded<d dZ Gdde Z!Gdde!Z"d!d"dZ# d# d$dZ$ d# d%dZ%d&dZ&d'dZ' d( d)dZ( d* d+dZ)d,dZ*d-dZ+y#e$rd ZYwxYw).aModule for serialization and deserialization of Albumentations transforms. This module provides functionality to serialize transforms to JSON or YAML format and deserialize them back. It implements the Serializable interface that allows transforms to be converted to and from dictionaries, which can then be saved to disk or transmitted over a network. This is particularly useful for saving augmentation pipelines and restoring them later with the exact same configuration. ) annotationsN)ABCABCMetaabstractmethod)MappingSequence)Enum)Path)AnyLiteralTextIO)warnTF) __version__) from_dictloadsaveto_dictzdict[str, SerializableMeta]SERIALIZABLE_REGISTRYNON_SERIALIZABLE_REGISTRYcd|jd}|dk(s|jds|S||dzdS)N.zalbumentations.)rfind startswith)class_fullname split_indexs l/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/albumentations/core/serialization.pyshorten_class_namer'sD &&s+Kb 9 9:K L +/+ ,,cZeZdZdZdfd ZeddZeddZed dZxZ S) SerializableMetazA metaclass that is used to register classes in `SERIALIZABLE_REGISTRY` or `NON_SERIALIZABLE_REGISTRY` so they can be found later while deserializing transformation pipeline using classes full names. ct||||g|i|}|dk7rHt|vr@|jr|t|j <|S|t |j <|S)N Serializable)super__new__ris_serializablerget_class_fullnamer)clsnamebasesargskwargscls_obj __class__s rr&zSerializableMeta.__new__8sr'/#tUDTDVD > !c&6&&(FM%g&@&@&BCKR)'*D*D*FGr cy)NFr)s rr'z SerializableMeta.is_serializableAsr ct|SN)get_shortest_class_fullnamer2s rr(z#SerializableMeta.get_class_fullnameEs *3//r ciSr4r1r2s r_to_dictzSerializableMeta._to_dictIs r ) r*strr+ztuple[type, ...]r,r r-r returnr"r9boolr9r8r9dict[str, Any]) __name__ __module__ __qualname____doc__r& classmethodr'r(r7 __classcell__)r/s@rr"r"3sI00r r"c`eZdZeeddZeeddZeddZd d dZy) r$ctr4NotImplementedErrorr2s rr'zSerializable.is_serializableO "!r ctr4rGr2s rr(zSerializable.get_class_fullnameTrIr ctr4rG)selfs rto_dict_privatezSerializable.to_dict_privateYs!!r c|dvrd|d} t| |j}t|d S#t$r=|dk(ri}tjd|d|j j dd YNwxYw) aTake a transform pipeline and convert it to a serializable representation that uses only standard python data types: dictionaries, lists, strings, integers, and floats. Args: self (Serializable): A transform that should be serialized. If the transform doesn't implement the `to_dict` method and `on_not_implemented_error` equals to 'raise' then `NotImplementedError` is raised. If `on_not_implemented_error` equals to 'warn' then `NotImplementedError` will be ignored but no transform parameters will be serialized. on_not_implemented_error (str): `raise` or `warn`. >rraisez(Unknown on_not_implemented_error value: z . Supported values are: 'raise' rOz2Got NotImplementedError while trying to serialize z;. Object arguments are not preserved. The transform class 'z`' needs to implement 'to_dict_private' or inherit from BasicTransform to be properly serialized. stacklevel)r transform) ValueErrorrMrHwarningsrr/r?r)rLon_not_implemented_errormsgtransform_dicts rrzSerializable.to_dict]s $+< <<=U) r?r@rArCrr'r(rMrr1r rr$r$NsR""""""Ir r$) metaclassc$|j|S)a~Take a transform pipeline and convert it to a serializable representation that uses only standard python data types: dictionaries, lists, strings, integers, and floats. Args: transform (Serializable): A transform that should be serialized. If the transform doesn't implement the `to_dict` method and `on_not_implemented_error` equals to 'raise' then `NotImplementedError` is raised. If `on_not_implemented_error` equals to 'warn' then `NotImplementedError` will be ignored but no transform parameters will be serialized. on_not_implemented_error (str): `raise` or `warn`. )r)rSrVs rrr}s   5 66r c|jdtvr=|d}|d|d} t||j|}|td|d|Sy)N__class_fullname__r?z6To deserialize a non-serializable transform with name z you need to pass a dict withz Non-serializable transform with z# was not found in `nonserializable`)getrrT)rSnonserializabler*rWresult_transforms rinstantiate_nonserializablerasy}})*.GG$  "J4&PmnC @S/ !*..t4  ?vEhij j r cpt|d}t||}|r|S|d}|jDcic]\}}|dk7s ||}}}d|vr|dvrtd|ddd |d<tt |}d |vr#|d D cgc]} t d| i| c} |d <|d i|Scc}}wcc} w) aArgs: transform_dict: A dictionary with serialized transform pipeline. nonserializable (dict): A dictionary that contains non-serializable transforms. This dictionary is required when you are restoring a pipeline that contains non-serializable transforms. Keys in that dictionary should be named same as `name` arguments in respective transforms from a serialized pipeline. rSr]p)Compose Sequentialz Transform z; has no 'p' parameter in serialized data, defaulting to 0.5rPrQg? transformsr_r1)register_additional_transformsraitemsrrrr) rXr_rSlmbdr*kvr,r)ts rrrs#${+I &y/ BD  ) *D&__. LTQ!7K2KAqD LD L $4'@@ z$Z [hijS  24 8 9Ctdhiudvw_`ia(8/Zw\ ;; Mxs B-B- B3c*|dvrtd|dy)N>jsonyamlzUnknown data_format z*. Supported formats are: 'json' and 'yaml')rT) data_formats rcheck_data_formatrrs&**/ }!$$==#x C)=+./aq!//"3-399636>/s B"Bct||j|}t|}t|tt frpt |j d5}|dk(r,ts d}t|tj||dn|dk(rtj||dddy|dk(r,ts d}t|tj||dy|dk(rtj||d yy#1swYyxYw) a Serialize a transform pipeline and save it to either a file specified by a path or a file-like object in either JSON or YAML format. Args: transform (Serializable): The transform pipeline to serialize. filepath_or_buffer (Union[str, Path, TextIO]): The file path or file-like object to write the serialized data to. If a string is provided, it is interpreted as a path to a file. If a file-like object is provided, the serialized data will be written to it directly. data_format (str): The format to serialize the data in. Valid options are 'json' and 'yaml'. Defaults to 'json'. on_not_implemented_error (str): Determines the behavior if a transform does not implement the `to_dict` method. If set to 'raise', a `NotImplementedError` is raised. If set to 'warn', the exception is ignored, and no transform arguments are saved. Defaults to 'raise'. Raises: ValueError: If `data_format` is 'yaml' but PyYAML is not installed. )rVwrpzr4)rSr>r_dict[str, Any] | Noner9Serializable | None)rXr>r_rr9r)rqLiteral['json', 'yaml']r9None)rwr r9r )rorO) rSr$rstr | Path | TextIOrqrrVzLiteral['raise', 'warn']r9r)roN)rrrqrr_rr9object)r9r)r)z type[Any]r9r8),rB __future__rimportlib.utilrrorUabcrrrcollections.abcrrenumr pathlibr typingr r r rrpr}rrr__all__r__annotations__rrr"r$rrarrrrurrrhr5r1r rrst# ,,-''N ' 268279;6; -w6,I-,I^ 7".2*&.2"*>i 7,29@ -@-@+-@)-@7 -@  -@d,2-1-F+-F(-F+-F -F`  .C NsCCC