Datasets
Models
Downloads: 1
111 lines (111 with data), 9.5 kB
3
Ř>»a7 Ń @ s× d dl Z d dljZd dlZd dljjZd dlm Z ddl
T d dl Z ddä Z G ddä dej
âZG d d
ä d
ej
âZG d d ä d ej
âZG d
dä dej
âZddä ZdS )Ú N)┌VariableÚ )┌*c C s |d d }|d dkr"|d }n|d }t jd|ůd|ůf }dt|d | â| dt|d | â| }t j| |||ft jdŹ}||t| ât|âddůddůf< tj|âjâ S )z1Make a 2D bilinear kernel suitable for upsamplingr Ú g Ó?Nr )┌dtype) ┌np┌ogrid┌abs┌zeros┌float64┌range┌torch┌
from_numpy┌float)┌ in_channels┌ out_channels┌ kernel_size┌factor┌centerZog┌filt┌weightę r ˙(/home/ws/yanghan/Newcode/model/Vggnet.py┌get_upsampling_weight
s
r c s* e Zd ZdZdç fddä Zddä Zç ZS )┌
DoubleConvz
Double Conv for U-Net
Ú c
s t t| âjâ t|â}t|â}tjtj||||dŹtj|âtjddŹtj||||dŹtj|âtjddŹâ| _ xĺ| j
â D ]ć}t |tjârĎ|j d |j d |j
}|jjjdtjd| ââ |jd k rŘ|jjjâ qvt |tjârv|jjjddâ |jjjdâ qvW d S ) N)┌paddingT)┌inplacer r g @g ?g{«Gßzö?)┌superr ┌__init__┌cal_same_padding┌nn┌
Sequential┌Conv2d┌ BatchNorm2d┌ReLU┌conv┌modules┌
isinstancer r r ┌data┌normal_┌math┌sqrt┌bias┌zero_┌fill_) ┌self┌in_ch┌out_ch┌k_1┌k_2┌ padding_1┌ padding_2┌m┌n)┌ __class__r r r s&
zDoubleConv.__init__c C s
| j |âS )N)r& )r0 ┌xr r r ┌forward6 s zDoubleConv.forward)r r )┌__name__┌
__module__┌ __qualname__┌__doc__r r; ┌
__classcell__r r )r9 r r s r c s6 e Zd Zç fddäZddä Zd
ddäZdd ä Zç ZS ) ┌VGG16Netc sB t t| âjâ || _|| _tjdddddŹ| _tjdddddŹ| _ tj
â | _ tjdddddŹ| _ tjdddddŹ| _
tj
â | _tjd d dd
Ź| _tjdd dddŹ| _tjd ddddŹ| _tj
â | _tjd d dddŹ| _tjd ddddŹ| _tj
â | _tjd d dd
Ź| _tjd d dddŹ| _tjd ddddŹ| _tj
â | _tjd d dddŹ| _tjd ddddŹ| _ tj
â | _ tjd d dddŹ| _ tjd ddddŹ| _ tj
â | _ tjd d dd
Ź| _!tjd d
dddŹ| _"tjd
ddddŹ| _#tj
â | _$tjd
d
dddŹ| _%tjd
ddddŹ| _&tj
â | _'tjd
d
dddŹ| _(tjd
ddddŹ| _)tj
â | _*tjd d dd
Ź| _+tjd
d
dddŹ| _,tjd
ddddŹ| _-tj
â | _.tjd
d
dddŹ| _/tjd
ddddŹ| _0tj
â | _1tjd
d
dddŹ| _2tjd
ddddŹ| _3tj
â | _4tj5d
d
d d dŹ| _6t7d
d ddŹ| _8t7d
d ddŹ| _9t7d
d ddŹ| _:t7d
d ddŹ| _;tj<â | _=tj>tj?dâtjd
d ddddŹtjd âtj
â â| _@tjdd dddŹ| _Atjd â| _Btjd ddddŹ| _Ctjdâ| _Dtj>tjdd dddddŹtjd âtj<â tjd d dddddŹtjd âtj
â â| _Etjd | jdddŹ| _Ftj| jddddŹ| _G| jHâ d S ) Nr Ú@ r )r g˝hŃłÁ°ń>gÜÖÖÖÖÖ╣?T)┌eps┌momentum┌affiner )┌stride┌ ceil_modeÚÇ Ú i )rF )┌rateÚ Ú Ú F)rF r- i )r- Ú0 i0 )r rF r r- )r rF )r r )Ir rA r Zn_class┌in_dimr! r# ┌conv1_1r$ ┌bn1_1┌PReLU┌relu1_1┌conv1_2┌bn1_2┌relu1_2┌ MaxPool2d┌pool1┌conv2_1┌bn2_1┌relu2_1┌conv2_2┌bn2_2┌relu2_2┌pool2┌conv3_1┌bn3_1┌relu3_1┌conv3_2┌bn3_2┌relu3_2┌conv3_3┌bn3_3┌relu3_3┌pool3┌conv4_1┌bn4_1┌relu4_1┌conv4_2┌bn4_2┌relu4_2┌conv4_3┌bn4_3┌relu4_3┌pool4┌conv5_1┌bn5_1┌relu5_1┌conv5_2┌bn5_2┌relu5_2┌conv5_3┌bn5_3┌relu5_3┌ConvTranspose2d┌up1┌ ASPP_module┌aspp1┌aspp2┌aspp3┌aspp4r% ┌relur" ┌AdaptiveAvgPool2d┌global_avg_pool┌conv1┌bn1┌conv2┌bn2┌ last_conv┌pred1┌conv_f1┌_initialize_weights)r0 rO ┌out_dim)r9 r r r ; sŐ
zVGG16Net.__init__c C sö xÄ| j â D ]é}t|tjârDtjj|jjddŹ |jd k rD|jjj â t|tj
âr
|j d |j d ksht ét
|j|j|j d â}|jjj|â q
W d S )Ng{«Gßzä?)┌stdr r )r' r( r! r# ┌initr* r r) r- r. r} r ┌AssertionErrorr r r ┌copy_)r0 r7 Zinitial_weightr r r rÄ ś s
z VGG16Net._initialize_weightsTc C sj | j |â}|}| j| j| j|âââ}| j| j| j|âââ}| j|â}|}| j| j | j
|âââ}| j | j | j
|âââ}| j|â}|}| j| j| j|âââ}| j| j| j|âââ}| j| j| j|âââ}|}| j|â}|}| j| j| j|âââ}| j | j | j |âââ}| j | j | j!|âââ}| j"|â}|} | j#| j$| j%|âââ}| j&| j'| j(|âââ}|}
|Ér\| j)|â}| j*|â} | j+|â} | | } | j,|â}
| |
}
| j-|â}|
| }| j.|â}t/j0|d
dddŹ}|| }t1j2| | |
||fddŹ}| j3| j4| j5|âââ}t/j0|d dddŹ}| j3| j6| j7|âââ}t1j2||fddŹ}| j8|â}t/j0|d dddŹ}| j9|â}|S |||
gS d S )
z Encoder rL ┌bilinearT)┌size┌mode┌
align_cornersr )┌dimÚ Ú` N)rL rL )rÖ rÖ )rÜ rÜ ):rŹ rS rQ rP rV rU rT rX r[ rZ rY r^ r] r\ r_ rb ra r` re rd rc rh rg rf ri rl rk rj ro rn rm rr rq rp rs ry rx rw r| r{ rz r~ rÇ rü ré râ rć ┌F┌ interpolater
┌caträ rł rç rŐ rë rő rî )r0 r: Zflag1┌h┌o1Zo2Zo3Z
low_level_feaZo4Zo5Zo6┌x1┌x2Zx3Zx4Zx5rî r r r r; ą s\
zVGG16Net.forwardc + C sľ | j | j| j| j| j| j| j| j| j| j | j
| j | j | j
| j| j| j| j| j| j| j| j| j| j| j| j| j| j| j | j | j | j | j | j!| j"| j#| j$| j%| j&| j'| j(| j)| j*g+}xÓt+|j,|âD ]đ\}}t-|t.j/âo▄t-|t.j/âÉr*|j0j1â |j0j1â ks°t2é|j3j1â |j3j1â kÉst2é|j0j4|j0_4|j3j4|j3_4t-|t.j5ârżt-|t.j5ârż|j0j1â |j0j1â kÉs\t2é|j3j1â |j3j1â kÉsvt2é|j0j4|j0_4|j3j4|j3_4qżW d S )N)6rP rQ rS rT rU rV rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs rt ru rv rw rx ry rz r{ r| ┌zip┌featuresr( r! r# r rĽ rĺ r- r) r$ )r0 Zvgg16_bnrú ┌l1┌l2r r r ┌copy_params_from_vgg16_bnš s8 z"VGG16Net.copy_params_from_vgg16_bn)T)r< r= r> r rÄ r; rŽ r@ r r )r9 r rA : s ]
BrA c s, e Zd Zç fddäZddä Zddä Zç ZS )r c sd t t| âjâ |dkr d}d}nd}|}tj|||d||ddŹ| _tj|â| _tjâ | _ | j
â d S )Nr r r F)r rF r ┌dilationr- ) r r r r! r# ┌atrous_convolutionr$ ┌bnr% rä ┌ _init_weight)r0 ZinplanesZplanesrJ r r )r9 r r r s
zASPP_module.__init__c C s | j |â}| j|â}| j|âS )N)rĘ rę rä )r0 r: r r r r; s
zASPP_module.forwardc C sZ xT| j â D ]H}t|tjâr,tjjj|jâ q
t|tjâr
|jj j
dâ |j j j â q
W d S )Nr )
r' r( r! r# r
rĹ ┌kaiming_normal_r r$ r) r/ r- r. )r0 r7 r r r r¬ s zASPP_module._init_weight)r< r= r> r r; r¬ r@ r r )r9 r r
s r c s( e Zd ZdZç fddäZddä Zç ZS )┌ ConvLSTMCellz,
Generate a convolutional LSTM cell
c s: t t| âjâ || _|| _tj|| d| dddŹ| _d S )NÚ r r )r )r rČ r ┌
input_size┌ hidden_sizer! r# ┌Gates)r0 r« r» )r9 r r r - s zConvLSTMCell.__init__c C s┌ |j jâ d }|j jâ dd ů }|d krV|| jgt|â }ttj|ââttj|ââf}|\}}tj||fdâ}| j|â} | j ddâ\}
} } }
tj
|
â}
tj
| â} tj
| â} tj |
â}
| | |
|
}| tj |â }||fS )Nr r r rş ) r) rĽ r» ┌listr r
r
rŁ r░ ┌chunk┌sigmoid┌tanh)r0 ┌input_Z
prev_state┌
batch_size┌ spatial_sizeZ
state_sizeZ prev_hiddenZ prev_cellZstacked_inputsZgatesZin_gateZ
remember_gateZout_gateZ cell_gate┌cell┌hiddenr r r r; 3 s"
zConvLSTMCell.forward)r< r= r> r? r r; r@ r r )r9 r rČ ( s rČ c C sľ t | jâ âd }|}tj||ddddŹ}tjd
d d gd
ddgdddggddŹ}|jdâ}tj||ddŹ}tj||d dŹ}tj |âj
â |j _ || â}|S )Nr r F)r r r- Ú ┌float32)r )┌axisr Ú rŻ rŻ rŻ rŻ rŻ rŻ rŻ )r r r r )
r▒ rĽ r! r# r ┌array┌reshape┌repeatr
r ┌cudar r) )┌im┌
in_channel┌ out_channelZconv_opZ sobel_kernelZ edge_detectr r r ┌EdgeConvZ s &
r┼ )r
┌torch.nnr! ┌numpyr Ztorch.nn.functional┌
functionalrŤ Ztorch.autogradr ┌_utilsr+ r ┌Moduler rA r rČ r┼ r r r r ┌<module> s
Q 2