Datasets
Models
Downloads: 1
75 lines (74 with data), 5.3 kB
3
²a ã @ s| d dl jZd dlZd dlmZ ddlT dddZG dd dejZG d d
d
ejZ G d d d ejZ
G d
d dejZ dS )é N)Úsqrté )Ú*c C s0 t jt j| ||||t j|ddt jd}|S )uÅ
conv-bn-relu for unet
:param in_channel: è¾å
¥éé
:param out_channel: è¾åºéé
:param kernel: å·ç§¯å
:param stride: æ¥é¿
:param padding:
:return:
gü©ñÒMbP?)ZepsT)ÚnnÚ
SequentialÚConv2dÚ BatchNorm2dÚReLU)Ú
in_channelZ out_channelZkernelÚstrideÚpaddingZlayer© r
ú&/home/ws/yanghan/Newcode/model/Unet.pyÚ conv_relu 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||||dtj|tjddtj||||dtj|tjdd| _ x| j
D ]}t |tjrÐ|j d |j d |j
}|jjjdtd| |jd k rú|jjj qvt |tjrv|jjjdd |jjjd qvW d S ) N)r T)Úinplacer r g @g ð?g{®Gáz?)Úsuperr Ú__init__Zcal_same_paddingr r r r r ÚconvÚmodulesÚ
isinstanceÚ kernel_sizeÚ out_channelsÚweightÚdataÚnormal_r ÚbiasÚzero_Úfill_) ÚselfZin_chZout_chZk_1Zk_2Z padding_1Z padding_2ÚmÚn)Ú __class__r
r r s&
zDoubleConv.__init__c C s
| j |S )N)r )r Úxr
r
r Úforward4 s zDoubleConv.forward)r r )Ú__name__Ú
__module__Ú __qualname__Ú__doc__r r% Ú
__classcell__r
r
)r# r r s r c s( e Zd ZdZ fddZdd Z ZS )Ú Inceptionu
ä¸ä¸ªå³æå³ç¨ç3è·¯inceptionm模å
è¾å
¥å°ºå¯¸åè¾åºå°ºå¯¸ä¸è´
Example:
>>> Inception(64,64//4,x,64//4,x,64//4,64//4)
c s t t| j t||d| _tjt||dt||ddd| _tjt||dt||ddd| _tjtj ddddt||d| _
dS )ul
å½ in_channel = out1_1 + out2_3 + out3_5 + out4_1æ¶
è¾å
¥åè¾åºå°ºå¯¸ä¸è´
r r )r é é )r r N) r r+ r r Ú branch1x1r r Ú branch3x3Ú branch5x5Ú MaxPool2dÚ branch_pool)r r
Zout1_1Zout2_1Zout2_3Zout3_1Zout3_5Zout4_1)r# r
r r > s
zInception.__init__c C sB | j |}| j|}| j|}| j|}tj||||fdd}|S )Nr )Údim)r. r/ r0 r2 ÚtorchÚcat)r r$ Úf1Úf2Zf3Zf4Úoutputr
r
r r% Z s
zInception.forward)r& r' r( r) r r% r* r
r
)r# r r+ 7 s r+ c s& e Zd Zd fdd Zdd Z ZS )ÚSELayeré c s\ t t| j tjd| _tjtj||| ddtjddtj|| |ddtj | _
d S )Nr F)r T)r ) r r9 r r ZAdaptiveAvgPool2dÚavg_poolr ZLinearr ZSigmoidÚfc)r ZchannelZ reduction)r# r
r r f s
zSELayer.__init__c C sF |j \}}}}| j|j||}| j|j||dd}||j| S )Nr )Úsizer; Úviewr< Z expand_as)r r$ ÚbÚcÚ_Úyr
r
r r% p s zSELayer.forward)r: )r& r' r( r r% r* r
r
)r# r r9 e s
r9 c s$ e Zd Z fddZdd Z ZS )ÚUnetc s¾ t t| j t|d| _tjd| _tddddddd| _ tdd| _
tjd| _ tdd| _ tjd| _
tdd| _tjd| _tdd| _tjddddd | _tdd| _tjddddd | _tdd| _tjddddd | _tdd| _tjddddd | _tdd| _tjd|d
| _x| j D ]}t |tjr|j d |j d
|j }|j j!j"d t#d | |j$d k r´|j$j!j% n,t |tj&r*|j j!j"d
d |j$j!j'd q*W d S )Né@ r- r: é0 é é i i )r r r g @g ð?g{®Gáz?)(r rC r r Úconv1r r1 Úpool1r+ Zres1Úconv2Úpool2Úconv3Úpool3Úconv4Úpool4Úconv5ZConvTranspose2dÚup6Úconv6Úup7Úconv7Úup8Úconv8Úup9Úconv9r Úconv10r r r r r r r r r r r r )r Zin_dimZout_dimr! r" )r# r
r r x s: z
Unet.__init__c C s | j |}| j|}| j|}| j|}| j|}| j|}| j|}| j|} | j| }
| j |
} t
j | |gdd} | j | }
| j
|
}t
j ||gdd}| j|}| j|}t
j ||gdd}| j|}| j|}t
j ||gdd}| j|}| j|}||fS )Nr )r3 )rH rI rJ rK rL rM rN rO rP rQ r4 r5 rR rS rT rU rV rW rX rY )r ÚinputsZc1Zp1Zc2Zp2Zc3Zp3Zc4Zp4Zc5Zup_6Zmerge6Zc6Zup_7Zmerge7Zc7Zup_8Zmerge8Zc8Zup_9Zmerge9Zc9Zc10r
r
r r% s.
z Unet.forward)r& r' r( r r% r* r
r
)r# r rC w s $rC )r r ) Ztorch.nnr r4 Zmathr Z_utilsr ZModuler r+ r9 rC r
r
r
r Ú<module> s
.