我现在已经用pycharm读取了相关的mat文件并为mat文件进行了加噪显示运行成功加噪成功,然后现在我想尝试着用DnCNN的方法为加噪后的文件进行去噪,我自己写了一次运行结果有报错然后问了一下相关专业的同学说是这是用GPU去噪(不太建议使用)比较建议用笔记本电脑的CPU直接处理去噪,请问一下有什么好的方法吗?
代码如下:
import scipy.io as scio
from scipy.io import loadmat
Testing=loadmat('C:/Users/13145/Desktop/Training_BrainImages_256x256_100.mat')
print(Testing)
import pandas as pd
clean_signal=pd.DataFrame([1,2],[3,4],columns=list('AB'),dtype=float)
print(clean_signal)
import numpy as np
mu,sigma=0,0.1
noise=np.random.normal(mu,sigma,[2,2])
print(noise)
signal=clean_signal+noise
print(signal)
signal.to_csv("output_Testing.csv",index=False)
import os
import numpy as np
import argparse
import tensorflow as tf
from model_DnCNN import DnCNN
os.environ=["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ=["CUDA_VISIBLE_DEVICES"]="0"
parser=argparse.ArgumentParser(decription="Tensorflow DnCNN Testing")
parser.add_argument("--epochs",default=50,type=int,help="Testing epochs")
parser.add_argument("--patch-size",default=40,type=int,help="patch size")
parser.add_argument("--c-dim",default=1,type=int,help="#of channels")
parser.add_argument("--batch-size",default=64,type=int,help="mini-batch size")
parser.add_argument("--lr",default=1e-3,type=float,help="Learning rate")
parser.add_argument("--lr-decay",default=30,type=int,help="Step of learning rate decay")
parser.add_argument("--weight-decay",default=1e-4,type=float,help="weight decay")
parser.add_argument("--sigma",default=25,type=int,help="noise level(default 25)")
parser.add_argument("--Testing-path",default='./desktop/Training_BrainImages_256x256_100.tfrecords',type=str,help="path to Testingset")
parser.add_argument("--validate-dir",default='C:/Users/13145',type=str,help="path to Testingset")
opt=parser.parse_args()
opt.checkpoint_path='./'+opt.modale_name+'_checkpoints'
def main():
model=DnCNN()
config=tf.ConfigProto()
config.gpu_options.allow_growth=True
with tf.Session(config=config) as sess:
model.Testing(sess,opt)
if name:='main':
tf.app.run
报错:C:\Users\13145\PycharmProjects\pythonProject1\venv\Scripts\python.exe "C:/Users/13145/Desktop/Test/full noise.py"
File "C:\Users\13145\Desktop\Test\full noise.py", line 28
os.environ=["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
SyntaxError: cannot assign to literal
Process finished with exit code 1