a PSico@sddlZddlZddlZddlmZmZddlmZddlmZm Z m Z m Z ddl m Z ddlmZgdZeGd d d ZGd d d e Zeed ddZe edddZdS)N) dataclassfields) signature)AnyCallableDictcast)StrEnum)load_state_dict_from_url) WeightsEnumWeights get_weightc@s2eZdZUdZeed<eed<eeefed<dS)r a This class is used to group important attributes associated with the pre-trained weights. Args: url (str): The location where we find the weights. transforms (Callable): A callable that constructs the preprocessing method (or validation preset transforms) needed to use the model. The reason we attach a constructor method rather than an already constructed object is because the specific object might have memory and thus we want to delay initialization until needed. meta (Dict[str, Any]): Stores meta-data related to the weights of the model and its configuration. These can be informative attributes (for example the number of parameters/flops, recipe link/methods used in training etc), configuration parameters (for example the `num_classes`) needed to construct the model or important meta-data (for example the `classes` of a classification model) needed to use the model. url transformsmetaN) __name__ __module__ __qualname____doc__str__annotations__rrrrrS/var/www/html/django/DPS/env/lib/python3.9/site-packages/torchvision/models/_api.pyr s r csheZdZdZedddZeeedddZe e e efdd d Z e d d d Z fddZZS)r aM This class is the parent class of all model weights. Each model building method receives an optional `weights` parameter with its associated pre-trained weights. It inherits from `Enum` and its values should be of type `Weights`. Args: value (Weights): The data class entry with the weight information. )valuecCs ||_dSN)_value_)selfrrrr__init__0szWeightsEnum.__init__)objreturncCsX|durTt|tur.|||jdd}n&t||sTtd|jd|jjd|S)N.z(Invalid Weight class provided; expected z but received )typerfrom_strreplacer isinstance TypeError __class__)clsrrrrverify3s  zWeightsEnum.verify)progressr cCst|j|dS)N)r+)r r)rr+rrrget_state_dict>szWeightsEnum.get_state_dict)r cCs|jjd|jS)Nr!)r(r_name_)rrrr__repr__AszWeightsEnum.__repr__cs6ttD] }|j|krt|j|Sqt|Sr)rr nameobject__getattribute__rsuper __getattr__)rr/fr(rrr3Ds  zWeightsEnum.__getattr__)rrrrr r classmethodrr*boolrrr,r.r3 __classcell__rrr5rr &s  r )r/r c Csz|d\}}Wn"ty4td|dYn0dtjtjddd}t|}|gddt |tj D}d}|D],}|j |d}|durt |tr|}qq|durtd|d ||S) z Gets the weight enum value by its full name. Example: "ResNet50_Weights.IMAGENET1K_V1" Args: name (str): The name of the weight enum entry. Returns: WeightsEnum: The requested weight enum. r!zInvalid weight name provided: 'z'.NcSs$g|]}|djdr|dqS)z __init__.py)__file__endswith).0xrrr ]szget_weight..zThe weight enum 'z0' for the specific method couldn't be retrieved.)split ValueErrorjoinsysmodulesr importlib import_moduleinspect getmembersismodule__dict__get issubclassr r$) r/ enum_name value_nameZbase_module_name base_moduleZ model_modules weights_enummZpotential_classrrrrLs$      r)fnr cCst|}d|jvrtdt|jdj}d}t|trHt|trH|}n(|jD] }t|trNt|trN|}qpqN|durtdt t|S)az Internal method that gets the weight enum of a specific model builder method. Might be removed after the handle_legacy_interface is removed. Args: fn (Callable): The builder method used to create the model. weight_name (str): The name of the weight enum entry of the specific model. Returns: WeightsEnum: The requested weight enum. weightsz-The method is missing the 'weights' argument.NzjThe WeightsEnum class for the specific method couldn't be retrieved. Make sure the typing info is correct.) r parametersrA annotationr&r#rLr __args__r)rRsigannrPtrrr_get_enum_from_fnns   rZ)rErGrC dataclassesrrrtypingrrrrZtorchvision._utilsr _internally_replaced_utilsr __all__r r rrrZrrrrs   &"