|h4dZddlmcmZddlmZddlmZddl m Z ddl m Z ddl mZddl mZdd lmZdd lmZd Zed zZed zZedzZedzZe Z d!dZedd d"dZedd d"dZ d#dZd$dZd$dZ d$dZ!edd$dZ"edd%dZ#ejHjKdejLejN e"_ejFje#_y)&aJNASNet-A models for TF-Keras. NASNet refers to Neural Architecture Search Network, a family of models that were designed automatically by learning the model architectures directly on the dataset of interest. Here we consider NASNet-A, the highest performance model that was found for the CIFAR-10 dataset, and then extended to ImageNet 2012 dataset, obtaining state of the art performance on CIFAR-10 and ImageNet 2012. Only the NASNet-A models, and their respective weights, which are suited for ImageNet 2012 are provided. The below table describes the performance on ImageNet 2012: --------------------------------------------------------------------------- Architecture | Top-1 Acc | Top-5 Acc | Multiply-Adds | Params (M) ---------------------|-----------|-----------|----------------|------------ NASNet-A (4 @ 1056) | 74.0 % | 91.6 % | 564 M | 5.3 NASNet-A (6 @ 4032) | 82.7 % | 96.2 % | 23.8 B | 88.9 Reference: - [Learning Transferable Architectures for Scalable Image Recognition]( https://arxiv.org/abs/1707.07012) (CVPR 2018) N)backend)imagenet_utils)training)VersionAwareLayers) data_utils) layer_utils) tf_logging) keras_exportzDhttps://storage.googleapis.com/tensorflow/keras-applications/nasnet/zNASNet-mobile.h5zNASNet-mobile-no-top.h5zNASNet-large.h5zNASNet-large-no-top.h5c |dvs4tjjj|s t d|dk(r|r| dk7r t dt |t r#d|vr|dk(rt dt|zdz| d } tj|| d tj|| }tjd k7r-tjd tjd d} nd} |tj!|}n/tj"|stj!||}n|}|d|dzzzdk7rt d|ztjdk(rdnd}|dz}tj%|dddddd|}tj'|ddd !|}d}t)||||dzzd"#\}}t)||||zd$#\}}t+|D]}t-|||d%|z#\}}t)||||zd&|z#\}}|s|n|}t+|D] }t-||||zd%||zdzz#\}}"t)||||dzzd&d|zz#\}}|s|n|}t+|D]&}t-||||dzzd%d|z|zdzz#\}}(tj/d'|}|rMtj1|}tj2| |tj5| | d()|}n=| d*k(rtj1|}n| d+k(rtj7|}|t9j:|}n|}t=j>||d,-}|dk(r| d.k(rO|rtAjBd/tDd0d12}ntAjBd3tFd0d42}|jI|nr| d k(rO|rtAjBd5tJd0d62}ntAjBd7tLd0d82}|jI|nt d9||jI|| rtj| |S):a%Instantiates a NASNet model. Reference: - [Learning Transferable Architectures for Scalable Image Recognition]( https://arxiv.org/abs/1707.07012) (CVPR 2018) For image classification use cases, see [this page for detailed examples]( https://keras.io/api/applications/#usage-examples-for-image-classification-models). For transfer learning use cases, make sure to read the [guide to transfer learning & fine-tuning]( https://keras.io/guides/transfer_learning/). Note: each TF-Keras Application expects a specific kind of input preprocessing. For NasNet, call `tf.keras.applications.nasnet.preprocess_input` on your inputs before passing them to the model. `nasnet.preprocess_input` will scale input pixels between -1 and 1. Args: input_shape: Optional shape tuple, the input shape is by default `(331, 331, 3)` for NASNetLarge and `(224, 224, 3)` for NASNetMobile. It should have exactly 3 input channels, and width and height should be no smaller than 32. E.g. `(224, 224, 3)` would be one valid value. penultimate_filters: Number of filters in the penultimate layer. NASNet models use the notation `NASNet (N @ P)`, where: - N is the number of blocks - P is the number of penultimate filters num_blocks: Number of repeated blocks of the NASNet model. NASNet models use the notation `NASNet (N @ P)`, where: - N is the number of blocks - P is the number of penultimate filters stem_block_filters: Number of filters in the initial stem block skip_reduction: Whether to skip the reduction step at the tail end of the network. filter_multiplier: Controls the width of the network. - If `filter_multiplier` < 1.0, proportionally decreases the number of filters in each layer. - If `filter_multiplier` > 1.0, proportionally increases the number of filters in each layer. - If `filter_multiplier` = 1, default number of filters from the paper are used at each layer. include_top: Whether to include the fully-connected layer at the top of the network. weights: `None` (random initialization) or `imagenet` (ImageNet weights) input_tensor: Optional TF-Keras tensor (i.e. output of `layers.Input()`) to use as image input for the model. pooling: Optional pooling mode for feature extraction when `include_top` is `False`. - `None` means that the output of the model will be the 4D tensor output of the last convolutional block. - `avg` means that global average pooling will be applied to the output of the last convolutional block, and thus the output of the model will be a 2D tensor. - `max` means that global max pooling will be applied. classes: Optional number of classes to classify images into, only to be specified if `include_top` is True, and if no `weights` argument is specified. default_size: Specifies the default image size of the model classifier_activation: A `str` or callable. The activation function to use on the "top" layer. Ignored unless `include_top=True`. Set `classifier_activation=None` to return the logits of the "top" layer. When loading pretrained weights, `classifier_activation` can only be `None` or `"softmax"`. Returns: A `keras.Model` instance. >NimagenetzThe `weights` argument should be either `None` (random initialization), `imagenet` (pre-training on ImageNet), or the path to the weights file to be loaded.r zWIf using `weights` as `"imagenet"` with `include_top` as true, `classes` should be 1000NzWhen specifying the input shape of a NASNet and loading `ImageNet` weights, the input_shape argument must be static (no None entries). Got: `input_shape=z`.K ) default_sizemin_size data_formatrequire_flattenweights channels_lastaThe NASNet family of models is only available for the input data format "channels_last" (width, height, channels). However your settings specify the default data format "channels_first" (channels, width, height). You should set `image_data_format="channels_last"` in your TF-Keras config located at ~/.keras/keras.json. The model being returned right now will expect inputs to follow the "channels_last" data format.channels_first)shape)tensorrrzwFor NASNet-A models, the `penultimate_filters` must be a multiple of 24 * (`filter_multiplier` ** 2). Current value: %drrrvalidF stem_conv1 he_normal)stridespaddinguse_biasnamekernel_initializerZڊ?MbP?stem_bn1axismomentumepsilonr&stem_1block_idstem_2z%dz reduce_%drelu predictions) activationr&avgmaxNASNetr&znasnet_mobile.h5models 020fb642bf7360b370c678b08e0adf61) cache_subdir file_hashznasnet_mobile_no_top.h5 1ed92395b5b598bdda52abe5c0dbfd63znasnet_large.h5 11577c9a518f0070763c2b964a382f17znasnet_large_no_top.h5 d81d89dc07e6e56530c4e77faddd61b5zDImageNet weights can only be loaded with NASNetLarge or NASNetMobile)'tfiogfileexists ValueError isinstancetuplestrrobtain_input_shaperimage_data_formatloggingwarningset_image_data_formatlayersInputis_keras_tensorConv2DBatchNormalization_reduction_a_cellrange_normal_a_cell ActivationGlobalAveragePooling2Dvalidate_activationDenseGlobalMaxPooling2Drget_source_inputsrModelrget_fileNASNET_MOBILE_WEIGHT_PATH NASNET_MOBILE_WEIGHT_PATH_NO_TOP load_weightsNASNET_LARGE_WEIGHT_PATHNASNET_LARGE_WEIGHT_PATH_NO_TOP) input_shapepenultimate_filters num_blocksstem_block_filtersskip_reductionfilter_multiplier include_topr input_tensorpoolingclassesrclassifier_activationold_data_format img_input channel_dimfiltersxpip0inputsmodel weights_paths _/var/www/html/test/engine/venv/lib/python3.12/site-packages/tf_keras/src/applications/nasnet.pyr8r8@s2x ) )RUU[[-?-?-H <  *D 0  ;& K  z ! 47:+6F GJN N   !33!--/# K  "o5 9  %%o6*LL{L3 &&|4 L LI$Ib$5q$89:a? D! "  0026FF!BK!R'G  &   A !! 64j "  A A  1g+Q./( DAq  1g**X DAq: BaGdajA1B  1g))K:4N EAr!aA :   ' 'Z!^a/0  1    #Q&&J/  EAr!aA :   '* *Q^a/!34  1  &!!$A  ) ) +A .**+@'J LL  5M    e --/2A  ))+A.A..|< NN618 4E* 3 )22&-!)@    *22-4!)@     | , S )22%,!)@    *22,3!)@     | ,#    7#%%o6 Lz&keras.applications.nasnet.NASNetMobilezkeras.applications.NASNetMobilec2t|ddddd|||||d| S)a Instantiates a Mobile NASNet model in ImageNet mode. Reference: - [Learning Transferable Architectures for Scalable Image Recognition]( https://arxiv.org/abs/1707.07012) (CVPR 2018) Optionally loads weights pre-trained on ImageNet. Note that the data format convention used by the model is the one specified in your TF-Keras config at `~/.keras/keras.json`. Note: each TF-Keras Application expects a specific kind of input preprocessing. For NASNet, call `tf.keras.applications.nasnet.preprocess_input` on your inputs before passing them to the model. Args: input_shape: Optional shape tuple, only to be specified if `include_top` is False (otherwise the input shape has to be `(224, 224, 3)` for NASNetMobile It should have exactly 3 inputs channels, and width and height should be no smaller than 32. E.g. `(224, 224, 3)` would be one valid value. include_top: Whether to include the fully-connected layer at the top of the network. weights: `None` (random initialization) or `imagenet` (ImageNet weights). For loading `imagenet` weights, `input_shape` should be (224, 224, 3) input_tensor: Optional TF-Keras tensor (i.e. output of `layers.Input()`) to use as image input for the model. pooling: Optional pooling mode for feature extraction when `include_top` is `False`. - `None` means that the output of the model will be the 4D tensor output of the last convolutional layer. - `avg` means that global average pooling will be applied to the output of the last convolutional layer, and thus the output of the model will be a 2D tensor. - `max` means that global max pooling will be applied. classes: Optional number of classes to classify images into, only to be specified if `include_top` is True, and if no `weights` argument is specified. classifier_activation: A `str` or callable. The activation function to use on the "top" layer. Ignored unless `include_top=True`. Set `classifier_activation=None` to return the logits of the "top" layer. When loading pretrained weights, `classifier_activation` can only be `None` or `"softmax"`. Returns: A TF-Keras model instance. Raises: ValueError: In case of invalid argument for `weights`, or invalid input shape. RuntimeError: If attempting to run this model with a backend that does not support separable convolutions. i rFrr: rerfrgrhrirjrrkrlrmrrnr8rdrjrrkrlrmrns rz NASNetMobileres9P  !3 r{z%keras.applications.nasnet.NASNetLargezkeras.applications.NASNetLargec2t|ddddd|||||d| S)a Instantiates a NASNet model in ImageNet mode. Reference: - [Learning Transferable Architectures for Scalable Image Recognition]( https://arxiv.org/abs/1707.07012) (CVPR 2018) Optionally loads weights pre-trained on ImageNet. Note that the data format convention used by the model is the one specified in your TF-Keras config at `~/.keras/keras.json`. Note: each TF-Keras Application expects a specific kind of input preprocessing. For NASNet, call `tf.keras.applications.nasnet.preprocess_input` on your inputs before passing them to the model. Args: input_shape: Optional shape tuple, only to be specified if `include_top` is False (otherwise the input shape has to be `(331, 331, 3)` for NASNetLarge. It should have exactly 3 inputs channels, and width and height should be no smaller than 32. E.g. `(224, 224, 3)` would be one valid value. include_top: Whether to include the fully-connected layer at the top of the network. weights: `None` (random initialization) or `imagenet` (ImageNet weights). For loading `imagenet` weights, `input_shape` should be (331, 331, 3) input_tensor: Optional TF-Keras tensor (i.e. output of `layers.Input()`) to use as image input for the model. pooling: Optional pooling mode for feature extraction when `include_top` is `False`. - `None` means that the output of the model will be the 4D tensor output of the last convolutional layer. - `avg` means that global average pooling will be applied to the output of the last convolutional layer, and thus the output of the model will be a 2D tensor. - `max` means that global max pooling will be applied. classes: Optional number of classes to classify images into, only to be specified if `include_top` is True, and if no `weights` argument is specified. classifier_activation: A `str` or callable. The activation function to use on the "top" layer. Ignored unless `include_top=True`. Set `classifier_activation=None` to return the logits of the "top" layer. When loading pretrained weights, `classifier_activation` can only be `None` or `"softmax"`. Returns: A TF-Keras model instance. Raises: ValueError: in case of invalid argument for `weights`, or invalid input shape. RuntimeError: If attempting to run this model with a backend that does not support separable convolutions. `Trrr~rrs rz NASNetLargers9P  !3 r{c tjdk(rdnd}tjd|5tj d|}|dk(r6tj t j||d||}d }nd }tj|||d ||d d |}tj|ddd||}tj d|}tj||d|d d d |}tj|ddd||}ddd|S#1swYSxYw)aIAdds 2 blocks of [relu-separable conv-batchnorm]. Args: ip: Input tensor filters: Number of output filters per layer kernel_size: Kernel size of separable convolutions strides: Strided convolution for downsampling block_id: String block_id Returns: A TF-Keras tensor rrrseparable_conv_block_r3rseparable_conv_1_pad_r$r&r sameseparable_conv_1_Fr")r#r&r$r%r'r(r)separable_conv_1_bn_r+separable_conv_2_)r&r$r%r'separable_conv_2_bn_N) rrK name_scoperOrW ZeroPadding2Dr correct_padSeparableConv2DrS)iprr kernel_sizer#r1rqrsconv_pads rz_separable_conv_blockrs0026FF!BK   3H:> ?'   f %b ) f $$&221kB,XJ7%AHH  " "  $XJ/* #     % %'z2 &     f %a (  " "  $XJ/* #     % %'z2 &  E' P HQ' P Hs C;D99Ec tjdk(rdnd}tjdk(rdnd}tj|}|tj|}tjd5||}n|||k7r?tjd|5t j d d | |}t j d d dd||}t j|dzd ddd|d|}t jd|} t jd| } t j d d dd|| } t j|dzd ddd|d| } tj|| g|}t j|ddd| |}dddn|||k7rtjd!|5t j d |}t j|d d dd"|dd#|}t j|ddd| |}dddddd|S#1swYxYw#1swYxYw#1swY|SxYw)$aAdjusts the input `previous path` to match the shape of the `input`. Used in situations where the output number of filters needs to be changed. Args: p: Input tensor which needs to be modified ip: Input tensor whose shape needs to be matched filters: Number of output filters to be matched block_id: String block_id Returns: Adjusted TF-Keras tensor rrrrN adjust_blockadjust_reduction_block_r3adjust_relu_1_r9rrrr adjust_avg_pool_1_r#r$r&rFadjust_conv_1_r")r$r%r&r')rrr)r$)rrr)croppingadjust_avg_pool_2_adjust_conv_2_)r,r(r) adjust_bn_r+adjust_projection_block_adjust_conv_projection_r#r$r&r%r') rrK int_shaperrOrWAveragePooling2DrRr Cropping2D concatenaterS) rtrrrr1rqimg_dimip_shapep_shapep1p2s rz _adjust_blockrSs0026FF!BK,,.2BBaG  $H}##A&   N +B 9A W '!2 2##&=hZ$HI* %%f^H:3N%O,,"#-hZ8 -   ]]qL"")(4'2 #))2B)CAF&&0@&A"E,,"#-hZ8 -   ]]qL"")(4'2 #&&Bxk&B--$# %hZ0 .  K* * X[ !W ,##&>xj$IJ %%f-a0MM""28*="'2"--$# %hZ0 .   eBF H}* * Z  eBF Hs>8+I4#DI)I4*A I( I4I% !I4(I1 -I44I>c tjdk(rdnd}tjd|5t||||}t j d|}t j |dddd|d d |}t j|d d d||}tjd5t||dd|}t||d|}tj||gd|}dddtjd5t||dd|} t||dd|} tj| | gd|} dddtjd5t jdddd| |} tj| |gd!|} dddtjd"5t jdddd#| |} t jdddd$| |}tj| |gd%|}dddtjd&5t||d'|}tj||gd(|}dddtj|  g|d)|*}ddd||fS#1swYxYw#1swYfxYw#1swYxYw#1swYxYw#1swYixYw#1swY|fSxYw)+zAdds a Normal cell for NASNet-A (Fig. 4 in the paper). Args: ip: Input tensor `x` p: Input tensor `p` filters: Number of output filters block_id: String block_id Returns: A TF-Keras tensor rrrnormal_A_block_r3rrnormal_conv_1_Fr"rr(r) normal_bn_1_r+block_1r normal_left1_)rr1normal_right1_r0 normal_add_1_r9Nblock_2 normal_left2_rnormal_right2_ normal_add_2_block_3 normal_left3_r normal_add_3_block_4 normal_left4_normal_right4_ normal_add_4_block_5 normal_left5_ normal_add_5_normal_concat_r,r&) rrKrrrOrWrRrSraddrr)rrtrrr1rqhx1_1x1_2x1x2_1x2_2x2x3x4_1x4_2x4x5rss rzrVrVs0026FF!BK   ohZ8 9K !R( 3   f %b ) MM  !(,*      % %z* &     * K("( 3 D )7~hZ%@DT4Lxj/IJB K   * K(7F}XJ-GD)7F~hZ-HDT4Lxj/IJB K   * F(($XJ/ )  B RGM(*DEB F   * K**$XJ/ +  D **%hZ0 +  D T4Lxj/IJB K   * F&7}XJ%?BRGM(*DEB  F    BB #!(,  OK X b5Lq K K K K F F K K F FCK X b5LsBK+7AJ,7K+AJ9K+2>K0K+ AK,K+ .K7)K+,J6 1K+9K >K+K K+K K+K( $K++K7c tjdk(rdnd}tjd|5t||||}t j d|}t j |dddd|d d |}t j|d d d||}t jtj|dd||}tjd5t||ddd|}t||ddd|}tj||gd|} dddtjd5t jdddd |!|} t||ddd"|} tj| | gd#|} dddtjd$5t jdddd%|!|} t||ddd&|}tj| |gd'|}dddtjd(5t jdddd)|! }tj |g}dddtjd*5t |dd)|+}t jdddd,|!|}tj||gd-|}dddtj g|d.|/}||fcdddS#1swYxYw#1swYtxYw#1swYxYw#1swYxYw#1swYhxYw#1swYyxYw)0zAdds a Reduction cell for NASNet-A (Fig. 4 in the paper). Args: ip: Input tensor `x` p: Input tensor `p` filters: Number of output filters block_id: String block_id Returns: A TF-Keras tensor rrrreduction_A_block_r3rrreduction_conv_1_Fr"rr(r)reduction_bn_1_r+rreduction_pad_1_rrrrreduction_left1_)r#r1)rreduction_right1_reduction_add_1_r9Nrrr reduction_left2_rreduction_right2_reduction_add_2_rreduction_left3_reduction_right3_reduction_add3_rreduction_left4_rr0reduction_right5_reduction_add4_reduction_concat_r)rrKrrrOrWrRrSrrrrr MaxPooling2Drr)rrtrrr1rqrh3rrrrrrx3_1x3_2rrx5_1x5_2rrss rzrTrT s0026FF!BK   0 ; <c !R( 3   f %b ) MM  $XJ/*      % %"8*- &    ! !"..q!4#H:."       * N(+H:6 D),XJ7 DT4L1A(/LMB N"   * N&&'z2 '  D ),XJ7 DT4L1A(/LMB N    * M**'z2 +  D ),XJ7 DT4L /KLB M    * &(('z2 )  B RH%B &   * M(GV0@ .KD&&( 3 '  D T4L /KLB M    R $XJ/   "uGcc0 N N" N N  M M  & & M MeccsB6L?*AL-L? AL L?8AL L?&9L'L?rs 0"! 4(2)*=9K.0BB#47P#P ,/@@"36N"N     #bJ ,.O   #SSl+-M   #SSn?C9 xY xZzqh:;< <==>=*>>EE 22  3 3F ,>>FFr{