/usr/local/lib64/python3.6/site-packages/caffe2/python/models/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/models/__pycache__/resnet.cpython-36.pyc (6369B)
3
Eg.
@ s d dl mZ d dlZG dd dZdddZdd d d d d dZddddgZej ejdZ e j
ej d!ddZ d"ddZ
dS )# )brewNc @ sH e Zd ZdZdddZddd Zd
d Zd d
ZdddZdddZ dS )
ResNetBuilderz8
Helper class for constructing residual blocks.
h㈵>?c C s< || _ d| _d| _|| _|| _|| _|| _|r2dnd| _d S )Nr )model
comp_countcomp_idx prev_blobis_test
bn_epsilon bn_momentumno_bias)selfr r
r r r r
r G/usr/local/lib64/python3.6/site-packages/caffe2/python/models/resnet.py__init__ s zResNetBuilder.__init__r r c
C sL | j d7 _ tj| j| jd| j| j f ||di f||||| jd | _| jS )Nr zcomp_%d_conv_%dMSRAFill) weight_initkernelstridegrouppadr )r r convr r
r r )r Z
in_filtersZ out_filtersr r r r r r r add_conv) s zResNetBuilder.add_convc C s t j| j| j| j| _| jS )N)r relur r
)r r r r add_reluB s
zResNetBuilder.add_reluc C s6 t j| j| jd| j| jf || j| j| jd| _| jS )Nzcomp_%d_spatbn_%d)epsilonmomentumr ) r
spatial_bnr r
r r r r
r )r num_filtersr r r add_spatial_bnJ s z ResNetBuilder.add_spatial_bnTc C s$ d| _ | j}| j||ddd |r,| j| | j | j||d||dd |rX| j| | j | j||dd}|r~| j|}||krtj| j|d| j ||di fd|| j d }|rtj
| j|d
| j || j | j | j
d }tj| j||gd | j| j f | _| j d7 _ | j | jd7 _|S )
Nr r )r r )r r r r )r zshortcut_projection_%dr )r r r r z shortcut_projection_%d_spatbn)r r r zcomp_%d_sum_%d)r r
r r! r r r r r r r r r
r sum) r
input_filtersZ base_filtersZoutput_filtersr r spatial_batch_norm
shortcut_blob last_convr r r add_bottleneckZ sd
z ResNetBuilder.add_bottleneckFc C s d| _ | j}| j||d|dkr"dnddd |r:| j| | j | j||ddd}|rb| j|}||krtj| j|d| j ||d i fd|dkrdnd| j d
}|rtj
| j|d | j |d | j d
}tj | j||gd| j| j f | _| j d7 _ | j | jd7 _d S )Nr r" Fr )r r r )r r zshortcut_projection_%dr )r r r r z shortcut_projection_%d_spatbngMbP?)r r zcomp_%d_sum_%d)
r r
r r! r r r r r r r r r# )r r$ r
down_samplingr% r& r' r r r add_simple_block sN
z ResNetBuilder.add_simple_blockN)r r )r r r )r r T)FT)
__name__
__module__ __qualname____doc__r r r r! r( r+ r r r r r s
P r Fc C s t j| |d|dddd t j| dddd|d t j| dd dd
d g}t| d d |d
}d}xjtd dD ]\} xNtd d| D ]<}
|j|
d kr|n|| || |
d kr| d krdndd qzW || }qfW t j| |jdddd t j | ddd | t j
| dd} | S )zp
Create residual net for smaller images (sec 4.2 of He et. al (2015))
num_groups = 'n' in the paper
conv1 r" r )r r Z conv1_spatbngMbP?)r r Zrelu1 @ r )r r r) TF)r* final_avg last_outsoftmax) r r r r r ranger+ average_poolr
fcr7 ) r datanum_input_channels
num_groups
num_labelsr filtersbuilderZ prev_filtersZgroupidxZblockidxr7 r r r create_resnet_32x32 s*
rA r) r" r5 $ ) " 2 e r Zresnext_builder h㈵>?c $ C s |t kr|jdj| t | }t}dddddg}|d krHddd ddg}|d! }tj| |d ||d d
i f| | d|
d
}tj| |d|d |||d}tj| ||}tj| |dddd
d}t | ||
|ddd}|| }xnt
dD ]b}|| } || } || } x
g?)r r r r
rB )r r r4 T)r r Zglobal_poolingz last_out_L{}r7 loss)rG rH )RESNEXT_BLOCK_CONFIGerrorformatRESNEXT_STRIDESr r r r max_poolr r8 r( r9 r
r: ZSoftmaxWithLossr7 )$r r; r< r>
num_layersr= num_width_per_grouplabelr no_lossr conv1_kernel conv1_stridefinal_avg_kernellogr r
Z
num_blocksstridesr Z num_featuresZ conv_blobZbn_blobZ relu_blobrX r@ Z inner_dimZ residual_idxZ residual_numZresidual_strideZdim_inZblk_idxr4 r6 r7 rR r r r create_resnext# sz
rb c C s$ t | |||ddd|||||| |
dS )NrI r r3 )
rY r= rZ r[ r r\ r r] r^ r_ )rb ) r r; r< r> r[ r r\ r r] r^ r_ r r r create_resnet50 s rc )F)r) r) r) r) )r" rB rC r" )r" rB rC r" )r" rB rD r" )r" r5 rE r" )r" rF rE r" )
NFFr rM r) rM NrN rO )NFFr rM r) rM )Z
caffe2.pythonr loggingr rA rT rW basicConfig getLoggerr` setLevelDEBUGrb rc r r r r s> Z
'
f