tensorflow v2 compat

CSFM.py

@@ -1,4 +1,4 @@
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from utils import gradient_sensitive_loss, tf_ms_ssim
 
 class Model(object):

ESPCN.py

@@ -1,4 +1,4 @@
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from utils import gradient_sensitive_loss, tf_ms_ssim
 
 class Model(object):

FSRCNN.py

@@ -1,4 +1,4 @@
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from utils import gradient_sensitive_loss, tf_ms_ssim
 
 class Model(object):

LapSRN.py

@@ -1,4 +1,4 @@
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import math
 from utils import bilinear_upsample_weights, bicubic_downsample
 

main.py

@@ -1,7 +1,7 @@
 from model import Model
 
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 import pprint
 import os

model.py

@@ -12,7 +12,7 @@ import importlib
 from random import randrange
 
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from PIL import Image
 import pdb

utils.py

@@ -9,7 +9,7 @@ import subprocess
 import io
 from random import randrange, shuffle
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from PIL import Image
 import numpy as np
 import multiprocessing