tensorflow.tools.graph_transforms missing? about edsr_tensorflow HOT 5 OPEN

docker run --gpus all -it --rm -v $PWD:/tmp -w /tmp tensorflow/tensorflow:1.14.0-gpu-py3 bash
pip install -U pip
pip install -r requirements.txt
python main.py --upscale --scale 4 --image 1.png

Traceback (most recent call last):
  File "main.py", line 2, in <module>
    import data_utils
  File "/tmp/data_utils.py", line 5, in <module>
    import cv2
  File "/usr/local/lib/python3.6/dist-packages/cv2/__init__.py", line 5, in <module>
    from .cv2 import *
ImportError: libGL.so.1: cannot open shared object file: No such file or directory

--- a/run.py
+++ b/run.py
@@ -302,12 +302,14 @@ class run:
 
             bicubic_image = cv2.resize(fullimg, None, fx=self.scale, fy=self.scale, interpolation=cv2.INTER_CUBIC)
 
-            cv2.imshow('Original image', fullimg)
-            cv2.imshow('EDSR upscaled image', HR_image)
-            cv2.imshow('Bicubic upscaled image', bicubic_image)
+            #cv2.imshow('Original image', fullimg)
+            #cv2.imshow('EDSR upscaled image', HR_image)
+            #cv2.imshow('Bicubic upscaled image', bicubic_image)
 
-            cv2.waitKey(0)
-            cv2.destroyAllWindows()
+            #cv2.waitKey(0)
+            #cv2.destroyAllWindows()
+            cv2.imwrite("./output.png", HR_image)
+            #print("path is:",path)

mkdir -p input; mkdir -p output; mkdir -p output/input
#put png's in "input" dir
find input -name '*.png'| sort | awk '{print "python main.py --upscale --scale 4 --image "$1" ; mv output.png output/"$s}'|bash