【文件属性】:
文件名称:高斯白噪声matlab代码-dncnn_pytorch:dncnn_pytorch
文件大小:143.57MB
文件格式:ZIP
更新时间:2021-05-26 15:36:16
系统开源
高斯白噪声matlab代码
新的培训和测试代码(和)
合并批量归一化(PyTorch)
import
torch
import
torch
.
nn
as
nn
def
merge_bn
(
model
):
'''
merge
all
'Conv+BN'
(or
'TConv+BN')
into
'Conv'
(or
'TConv')
based
on
https://github.com/pytorch/pytorch/pull/901
by
Kai
Zhang
(cskaizhang@gmail.com)
https://github.com/cszn/DnCNN
01/01/2019
'''
prev_m
=
None
for
k
,
m
in
list
(
model
.
named_children
()):
if
(
isinstance
(
m
,
nn
.
BatchNorm2d
)
or
isinstance
(
m
,
nn
.
BatchNorm1d
))
and
(
isinstance
(
prev_m
,
nn
.
Conv2d
)
or
isinstan