|h%2ddlZddlmZdZGddeZy)N) LooseVersionuIván de Paz CentenoceZdZdZdZdZedZedededefdZ ed Z d e d e fd Z d e d efdZ dd e dededede dededede fdZdd e de fdZ dd e dedede fdZdd e dede fdZdde fdZy) LayerFactoryz< Allows to create stack layers for a given network. )SAMEVALIDc||_yN)_LayerFactory__network)selfnetworks R/var/www/html/test/engine/venv/lib/python3.12/site-packages/mtcnn/layer_factory.py__init__zLayerFactory.__init__'s  c\|tjvrtdj|y)NzPadding {} not valid)rAVAILABLE_PADDINGS Exceptionformat)paddings r __validate_paddingzLayerFactory.__validate_padding*s, ,99 9299'BC C :rchannels_inputchannels_outputgroupcP||zdk7r td||zdk7r tdy)NrzrN kernel_size stride_sizerbiasedreluinput_layer_namec |j|jj| } t| j d} |j | ||fd} t jjj|5} |jd|d|d| |z|g}| | |}|r3|jd|g}t jj||}|r+t jj|| j}d d d |jj| y #1swY'xYw) uA Creates a convolution layer for the network. :param name: name for the layer :param kernel_size: tuple containing the size of the kernel (Width, Height) :param channels_output: ¿? Perhaps number of channels in the output? it is used as the bias size. :param stride_size: tuple containing the size of the stride (Width, Height) :param padding: Type of padding. Available values are: ('SAME', 'VALID') :param group: groups for the kernel operation. More info required. :param biased: boolean flag to set if biased or not. :param relu: boolean flag to set if ReLu should be applied at the end of the layer or not. :param input_layer_name: name of the input layer for this layer. If None, it will take the last added layer of the network. rc^tjj||ddddgS)Nrr)r7filtersstridesr)r#nnconv2d) input_valkernelrr@s r z'LayerFactory.new_conv..ys5RUU\\ KNKNA>.:.!rweightsrrr,biasesr+Nr8)_LayerFactory__validate_paddingr get_layerr"r _LayerFactory__validate_groupingr#r0r1variable_scope_LayerFactory__make_varrHbias_addrBr+r<)r r+r?rr@rrrArBrCr%rconvolvescoperKoutputrOs `` r new_convzLayerFactory.new_conv\s.$ (nn../?@ [224R89   %H! YY\\ ( ( . =%__Y{1~{ST~WeinWnp7A_BFk62FO3DE7F< =   F ; = =s B D??Ec|jj|}tjjj |5t |jd}|jd|g}tjj|tj|tjj|  z}ddd|jj|y#1swY'xYw)z Creates a new prelu layer with the given name and input. :param name: name for this layer. :param input_layer_name: name of the layer that serves as input for this one. ralpharNNr8) r rRr#r0r1rTr"rrUrHrBmultiplyr<)r r+rCr%rr\rYs r new_preluzLayerFactory.new_prelus nn../?@ YY\\ ( ( . ] !6!6!8!<=NOOGN3CODEUUZZ ,r{{5255::{lC[B[/\\F ]   F ;  ] ]s BC11C:c|j||jj|}tjj |d|d|ddgd|d|ddg||}|jj ||y)a Creates a new max pooling layer. :param name: name for the layer. :param kernel_size: tuple containing the size of the kernel (Width, Height) :param stride_size: tuple containing the size of the stride (Width, Height) :param padding: Type of padding. Available values are: ('SAME', 'VALID') :param input_layer_name: name of the input layer for this layer. If None, it will take the last added layer of the network. rr)r7ksizerGrr+r8N)rQr rRr#rH max_pool2dr<)r r+r?r@rrCr%rYs r new_max_poolzLayerFactory.new_max_pools (nn../?@ !! '(+a.+a.!&L)*KNKNA(N(/%) "+   F ;r output_countc>tjjj|5|jj |}|j |\}}|jd||g}|jd|g} |r.tjjjjn-tjjjj} | ||| |} ddd|jj| y#1swY'xYw)a Creates a new fully connected layer. :param name: name for the layer. :param output_count: number of outputs of the fully connected layer. :param relu: boolean flag to set if ReLu should be applied at the end of this layer. :param input_layer_name: name of the input layer for this layer. If None, it will take the last added layer of the network. rMrNrOrPNr8) r#r0r1rTr rRr*rUrH relu_layer xw_plus_br<) r r+rcrBrCr%r) dimensionrMrO operationfcs r new_fully_connectedz LayerFactory.new_fully_connectedsYY\\ ( ( . I..223CDK*.*>*>{*K ' iooi <7PoQG__Xl^_DF6: 22 @Y@YI+Wf4HB I   B 7 I Is CDDc"|jj|}ttjtdkrItj ||d}tj ||z }tj||d}nHtj ||d}tj ||z }tj||d}tjj|||}|jj||y)z Creates a new softmax layer :param name: name to set for the layer :param axis: :param input_layer_name: name of the input layer for this layer. If None, it will take the last added layer of the network. z1.5.0T) input_tensoraxiskeepdimsr8N) r rRrr# __version__ reduce_maxexp reduce_summathdivider<) r r+rmrCr%max_axis target_exp normalizesoftmaxs r new_softmaxzLayerFactory.new_softmaxsnn../?@  ',w*? ?}}+DSWXH h 67J :DSWXI}}+DSWXH h 67J :DSWXI''..Y=   G boolrZr^rbrjryrr rr sK+!DD]C]#]c]]%% =s=4=?S?u?39Y]1rs 4* # B=6B=r