Commit 16f8f132 authored by Mano Brabant's avatar Mano Brabant
Browse files

Remove Tensorflow

parent 1bbc378f
#-*- coding: utf-8 -*- #-*- coding: utf-8 -*-
# #
# This file is part of DnCnn4Holo. # This file is part of DnCnn4Holo.
...@@ -37,7 +36,6 @@ import sys ...@@ -37,7 +36,6 @@ import sys
import re import re
import pathlib import pathlib
import numpy as np import numpy as np
import tensorflow as tf
from PIL import Image from PIL import Image
from scipy.io import loadmat, savemat from scipy.io import loadmat, savemat
from glob import glob from glob import glob
...@@ -593,20 +591,6 @@ def cal_std_phase(im1, im2): ...@@ -593,20 +591,6 @@ def cal_std_phase(im1, im2):
return dev return dev
<<<<<<< HEAD
def tf_psnr(im1, im2):
'''
this function is deprecated
'''
# assert pixel value range is 0-1
#mse = tf.losses.mean_squared_error(labels=im2 * 255.0, predictions=im1 * 255.0)
mse = tf.losses.mean_squared_error(labels=im2, predictions=im1)
return 10.0 * (tf.log(1 / mse) / tf.log(10.0))
=======
>>>>>>> cfbb377a755bc7b5e2b9ead857e01e5576ec3f67
class NNRegressor(nt.NeuralNetwork): class NNRegressor(nt.NeuralNetwork):
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment