/usr/local/lib64/python3.6/site-packages/caffe2/python/models/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/models/__pycache__/shufflenet.cpython-36.pyc (4416B)
3
Eg @ s^ d dl mZ dddddgdddd dgdd
d d dgdd
dddgdZG dd dZdddZdS ) )brew 0 ` i t i i` i i i i )z0.5xz1.0xz1.5xz2.0xc @ sL e Zd ZdddZdd Zdd d
Zdd d
ZdddZdd Zdd Z dS )ShuffleNetV2Builder 1.0xFh㈵>c
C sJ || _ || _|| _|| _|| _t| | _dddg| _|| _|| _ | | _
d S )N ) model prev_blobnum_input_channels
num_labels
num_groupsOUTPUT_CHANNELSoutput_channels
stage_repeatsis_test detection
bn_epsilon)
selfr datar r r widthr r r r K/usr/local/lib64/python3.6/site-packages/caffe2/python/models/shufflenet.py__init__ s
z ShuffleNetV2Builder.__init__c
C s0 | j d }tj| j| jd| j|di fddd| _tj| j| jdddd| _xtt| j d d
| j D ]b\}\}}d j
|d d}| j ||| |}x2t |D ]&}d j
|d d |d }| j
|| qW qfW tj| j| jd
|| j d
d d| _| jj| jddd| _tj| j| jdj
| j| j d
| j| _d S )Nr Z stage1_convMSRAFillr r ) weight_initkernelstrideZ stage1_pool)r$ r% zstage{}_stride{}zstage{}_stride{}_repeat{}Zconv5)r$ avg_poolr z last_out_L{})r r convr r r Zmax_pool enumeratezipr formatadd_spatial_ds_unitrangeadd_basic_unitZ last_convZ AveragePoolr( Zfcr last_out)r in_channelsidx out_channelsZ n_repeatsprefixir r r create/ s4
"
zShuffleNetV2Builder.createc C s | j }}|d }| jr,| j||d ||}| j||d ||}| j||d ||}| jrn| j||d ||}| j||d ||}| j||d ||}| j||d ||}tj| j||g|d | _ | jjj | j |d
| j
d d | _ d S )
Nr Z_left_detectionZ left_dwconvZ _left_conv1_right_detection _right_conv1
_right_dwconv _right_conv2_concat _ch_shuffler& )groupr$ ) r r add_detection_unitadd_dwconv3x3_bnadd_conv1x1_bnr concatr netChannelShuffler )r r4 r1 r3 r% rightleftr r r r- R s.
z'ShuffleNetV2Builder.add_spatial_ds_unitr& c C s |d }|d }|d }| j jj| j||g | jrH| j||d ||}| j||d ||}| j||d ||}| j||d ||}tj | j ||g|d | _| j jj
| j|d | j d
d | _d S ) Nr _left_rightr7 r8 r9 r: r; r< r& )r= r$ ) r rB ZSplitr r r> r@ r? r rA rC r )r r4 r1 r% rE rD r r r r/ s s$
z"ShuffleNetV2Builder.add_basic_unitr c C sJ t j| j||d |||di f||d }t j| j||d || j| jd}|S )NZ_convr" )r$ r# r= pad_bn)epsilonr )r r) r
spatial_bnr r )r r r4 r1 r3 r$ rH Zout_blobr r r r> s
z&ShuffleNetV2Builder.add_detection_unitc C sR t j| j||||ddi fd}t j| j||d || j| jd}t j| j||}|S )Nr& r" )r$ r# rI )rJ r )r r) r rK r r Zrelu)r r blobr1 r3 r r r r@ s
z"ShuffleNetV2Builder.add_conv1x1_bnc C sH t j| j||||ddi f||dd
}t j| j||d || j| jd}|S )Nr r" r& )r$ r# r% r= rH rI )rJ r )r r) r rK r r )r r rL Zchannelsr% r r r r? s
z$ShuffleNetV2Builder.add_dwconv3x3_bnN)r r
FFr )r )r& )r r& )
__name__
__module__ __qualname__r! r6 r- r/ r> r@ r? r r r r r s
#
!
r NFc
C sR t | ||||d}|j |r$|jS |d k rN| j|j|gddg\}} || fS d S )N)r softmaxloss)r r6 r0 ZSoftmaxWithLoss)
r r r r labelr Zno_lossZbuilderrP rQ r r r create_shufflenet s rS )NFF)Z
caffe2.pythonr r r rS r r r r
s