UGS-Net / Git / [cf6a9e] /model/__pycache_

Models:
RichardZick/
UGS-Net
Downloads: 1
[cf6a9e]: / model / __pycache__ / Mymodel2.cpython-36.pyc
History
Download this file
127 lines (127 with data), 10.8 kB

3

¨aÕFã@s¦ddlZddljZddlZddljjZddlm	Z	ddl
TddlZddlTddZ
GdddejZGd	d
d
ejZGdddejZGd
ddejZddZdS)éN)ÚVariableé)Ú*cCs¶|dd}|ddkr"|d}n|d}tjd|d|f}dt|d||dt|d||}tj||||ftjd}||t|t|ddddf<tj|jS)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ÚcenterÚogÚfiltÚweight©rú*/home/ws/yanghan/Newcode/model/Mymodel2.pyÚget_upsampling_weights

 rcs*eZdZdZdfdd	ZddZZS)Ú
DoubleConvz
    Double Conv for U-Net
    éc	
stt|jt|}t|}tjtj||||dtj|tjddtj||||dtj|tjdd|_	x|j
D]}t|tjrÒ|jd|jd|j
}|jjjdtjd||jdk	rü|jjjqvt|tjrv|jjjdd|jjjdqvWdS)	N)ÚpaddingT)Úinplacerrg@gð?g{®Gáz?)ÚsuperrÚ__init__Úcal_same_paddingÚnnÚ
SequentialÚConv2dÚBatchNorm2dÚReLUÚconvÚmodulesÚ
isinstancerrrÚdataÚnormal_ÚmathÚsqrtÚbiasÚzero_Úfill_)	ÚselfÚin_chÚout_chÚk_1Úk_2Ú	padding_1Ú	padding_2ÚmÚn)Ú	__class__rrr s&

zDoubleConv.__init__cCs
|j|S)N)r')r1ÚxrrrÚforward6szDoubleConv.forward)rr)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r r<Ú
__classcell__rr)r:rrsrcs6eZdZfddZddZdddZd	d
ZZS)ÚMymodelcsztt|j||_||_tjddddd|_tjddddd|_	tj
|_tjddddd|_tjddddd|_
tj
|_tjd	d	dd
|_tjddddd|_tjddddd|_tj
|_tjddddd|_tjddddd|_tj
|_tjd	d	dd
|_tjddddd|_tjddddd|_tj
|_tjddddd|_tjddddd|_tj
|_tjddddd|_tjddddd|_tj
|_ tjd	d	dd
|_!tjdd
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
ddd|_8t7d
ddd|_9t7d
ddd|_:t7d
ddd|_;tj<|_=tj>tj?dtjd
dddddtjdtj
|_@tjddddd|_Atjd|_Btjddddd|_Ctjd|_Dtj>tjdddddddtjdtj<tjdddddddtjdtj
|_Etjd|jddd|_Ftjd|jddd|_Gtjd|jddd|_Htjd|jddd|_Itj|jdddd|_Jtjddddd|_Ktjddddd|_Ltjddddd|_Mtjd
dddd|_Ntjd|jdddd|_Otjddddd|_Ptjd|jddd|_Qtjd
dddd|_RtSddddd|_Ttjd
dddd|_Utjddddd|_Vtjddddd|_W|jXdS)Nré@r)rgñhãµøä>g¹?T)ÚepsÚmomentumÚaffiner)ÚstrideÚ	ceil_modeééi)rG)ÚrateéééF)rGr.i)r.é0i0)rrGrr.)rrGi)rr)YrrBr Ún_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Úpred2Úpred3Úpred4Úconv_f1Úconv_f2Úconv_channelalig1Úconv_channelalig2Úconv_channelalig3Zconv_fusion1Úconv_fusion2Úconv_fusion3Ú
conv_feature0ZGConvÚ
conv_gaborÚconv_boundary1Úconv_boundary2Ú
conv_feature1Ú_initialize_weights)r1rQÚout_dim)r:rrr :s¨













zMymodel.__init__cCsx|jD]}t|tjrDtjj|jjdd|jdk	rD|jjj	t|tj
r
|jd|jdkshtt
|j|j|jd}|jjj|q
WdS)Ng{®Gáz?)Ústdrr)r(r)r"r$Úinitr+rr*r.r/rrÚAssertionErrorrrrÚcopy_)r1r8Úinitial_weightrrrr¬s
zMymodel._initialize_weightsFTc$Cst|dkr8x|D]}tj||fdd}qW|j|}n
|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-|}t.j/|	dddd}|j0|}t.j/|
dddd}|j1|}|rÜ|j2|}|j3|}|j4|}||}|j5|}||}|j6|}||}|j7|}t.j/|d
ddd}||}tj|||||fdd}|j8|j9|j:|}t.j/|dddd}t.j/|dddd}|j8|j;|j<|
}
tj||
fdd}|j=|}t.j/|dddd}|}|j>|}|j?|}|j@|}tj||fdd}|jA|}tB|}|jC|}tj||fdd} |jD| } |jE| }!tj|||fdd}|jF|}tjG|||}"|"jHd|jIdddg}"|"|}tj|||fdd}|jJ|}|jK|}#||||#|!gS|rü|||	|||
g|||gfS|||	|||
gSd
S)z	 Encoder rr)Údimé`ÚbilinearT)ÚsizeÚmodeÚ
align_cornersrMéN)r¦r¦)r¦r¦)rMrM)r«r«)r¦r¦)r¦r¦)LÚlenr
ÚcatrrrUrSrRrXrWrVrZr]r\r[r`r_r^rardrcrbrgrfrerjrirhrkrnrmrlrqrprortrsrrrur{rzryr~r}r|rÚFÚinterpolaterrrrrrrrrrrrrrrrrrÚEdgeConvrrrrÚsigmoidÚrepeatÚshaperr)$r1r;Úcoarse_predÚflag1Úflag2ÚyÚhÚo1Úo2Úo3Ú
low_level_feaÚo4Úo5Úo6Úlow_ckdÚo3_2Úmid_ckdÚo6_2Úhigh_ckdÚx1Úx2Úx3Úx4Úx5Úmap1Úmap2rrrÚfeature_mapÚedge_mapZ	gabor_mapZedge_map_outÚ	edge_predZuncertainty_mapÚpredrrrr<¹s

























zMymodel.forwardc+Cs|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|j0j1ksøt2|j3j1|j3j1kst2|j0j4|j0_4|j3j4|j3_4t-|t.j5r¾t-|t.j5r¾|j0j1|j0j1ks\t2|j3j1|j3j1ksvt2|j0j4|j0_4|j3j4|j3_4q¾WdS)N)6rRrSrUrVrWrXrZr[r\r]r^r_r`rarbrcrdrerfrgrhrirjrkrlrmrnrorprqrrrsrtrurvrwrxryrzr{r|r}r~ÚzipÚfeaturesr)r"r$rr¨r¢r.r*r%)r1Úvgg16_bnrÑÚl1Úl2rrrÚcopy_params_from_vgg16_bn:s8z!Mymodel.copy_params_from_vgg16_bn)FT)r=r>r?r rr<rÕrArr)r:rrB9s
r
rBcs,eZdZfddZddZddZZS)rc	sdtt|j|dkr d}d}nd}|}tj|||d||dd|_tj||_tj|_	|j
dS)NrrrF)rrGrÚdilationr.)rrr r"r$Úatrous_convolutionr%Úbnr&rÚ_init_weight)r1ÚinplanesÚplanesrKrr)r:rrr _s
zASPP_module.__init__cCs|j|}|j|}|j|S)N)r×rØr)r1r;rrrr<ms

zASPP_module.forwardcCsZxT|jD]H}t|tjr,tjjj|jq
t|tjr
|jj	j
d|jj	jq
WdS)Nr)
r(r)r"r$r
r¡Úkaiming_normal_rr%r*r0r.r/)r1r8rrrrÙsszASPP_module._init_weight)r=r>r?r r<rÙrArr)r:rr^srcs(eZdZdZfddZddZZS)ÚConvLSTMCellz,
    Generate a convolutional LSTM cell
    cs:tt|j||_||_tj||d|ddd|_dS)Nérr)r)rrÝr Ú
input_sizeÚhidden_sizer"r$ÚGates)r1rßrà)r:rrr szConvLSTMCell.__init__cCsÚ|jjd}|jjdd}|dkrV||jgt|}ttj|ttj|f}|\}}tj||fd}|j|}	|	j	dd\}
}}}
tj
|
}
tj
|}tj
|}tj|
}
|||
|
}|tj|}||fS)NrrrrÞ)r*r¨ràÚlistrr
r
rráÚchunkr±Útanh)r1Úinput_Ú
prev_stateÚ
batch_sizeÚspatial_sizeÚ
state_sizeÚprev_hiddenÚ	prev_cellÚstacked_inputsÚgatesÚin_gateÚ
remember_gateÚout_gateÚ	cell_gateÚcellÚhiddenrrrr<s"




zConvLSTMCell.forward)r=r>r?r@r r<rArr)r:rrÝ|srÝcCst|jd}|}tj||dddd}tjd
ddgd
ddgdddggdd}|jd}tj||dd}tj||d	d}tj	|j
|j_||}|S)NrrF)rrr.éÚfloat32)r)Úaxisréÿÿÿÿr÷r÷r÷r÷r÷r÷r÷)rrrr)
râr¨r"r$rÚarrayÚreshaper²r
rÚcudarr*)ÚimÚ
in_channelÚout_channelÚconv_opÚsobel_kernelÚedge_detectrrrr°®s&
r°)r
Útorch.nnr"ÚnumpyrZtorch.nn.functionalÚ
functionalr®Ztorch.autogradrÚ_utilsr,ZGConv.gcn.modules.GConvrÚModulerrBrrÝr°rrrrÚ<module>s
'2