Commit 96f42832 authored by Marie Tahon's avatar Marie Tahon
Browse files

remove torchvision from data.py and ref to tf in utils

parent a7eff5a3
import os
import torch
import torch.utils.data as td
import torchvision as tv
#import torchvision as tv
import numpy as np
from PIL import Image
from utils import *
......
......@@ -368,10 +368,6 @@ class Experiment(object):
def trace(self):
loss_tab = []
for k,v in (self.history):
loss_tab = np.append(loss_tab,round(k['loss'],6))
print("affichage graphique loss: ")
plt.plot(np.arange(0,len(loss_tab)),loss_tab)
plt.title("Losses/epoch Graph ")
......
......@@ -37,7 +37,7 @@ import sys
import re
import pathlib
import numpy as np
import tensorflow as tf
#import tensorflow as tf
from PIL import Image
from scipy.io import loadmat, savemat
from glob import glob
......@@ -594,6 +594,9 @@ def cal_std_phase(im1, im2):
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)
......
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