
    i0                     d    S SK r S SKrS SKJrJrJrJr  SSKJr   " S S\5      r	 " S S\5      r
g)	    N)resnet18ResNet18_Weightsresnet50ResNet50_Weights   )EagerModelBasec                   V    \ rS rSrS rS\R                  R                  4S jrS r	Sr
g)ResNet18Model   c                     g N selfs    f/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/examples/models/resnet/model.py__init__ResNet18Model.__init__           returnc                     [         R                  " S5        [        [        R                  S9n[         R                  " S5        U$ )Nz"Loading torchvision resnet18 modelweightsz!Loaded torchvision resnet18 model)logginginfor   r   IMAGENET1K_V1)r   resnet18_models     r   get_eager_modelResNet18Model.get_eager_model   3    9:!*:*H*HI89r   c                 4    Sn[         R                  " U5      4$ N)         r%   torchrandnr   input_shapes     r   get_example_inputs ResNet18Model.get_example_inputs       &K(**r   r   N__name__
__module____qualname____firstlineno__r   r'   nnModuler   r+   __static_attributes__r   r   r   r
   r
            +r   r
   c                   V    \ rS rSrS rS\R                  R                  4S jrS r	Sr
g)ResNet50Model$   c                     g r   r   r   s    r   r   ResNet50Model.__init__%   r   r   r   c                     [         R                  " S5        [        [        R                  S9n[         R                  " S5        U$ )Nz"Loading torchvision resnet50 modelr   z!Loaded torchvision resnet50 model)r   r   r   r   r   )r   resnet50_models     r   r   ResNet50Model.get_eager_model(   r    r   c                 4    Sn[         R                  " U5      4$ r"   r&   r)   s     r   r+    ResNet50Model.get_example_inputs.   r-   r   r   Nr.   r   r   r   r8   r8   $   r6   r   r8   )r   r'   torchvision.modelsr   r   r   r   
model_baser   r
   r8   r   r   r   <module>rC      s4       (+N ++N +r   