change cv to imageio

.gitignore

@@ -1,4 +1,5 @@
 __pycache__
 *.log
 *.pdparams
-*.mp4
+*.mp4
+*.gif

README.md

@@ -8,6 +8,19 @@ app_file: gradio_painttransformer.py
 pinned: false
 ---
 
+- 2021-12-21 first created
+    - opencv error:
+        ```
+        Traceback (most recent call last):
+        File "gradio_painttransformer.py", line 12, in <module>
+            import cv2
+        File "/home/user/.local/lib/python3.8/site-packages/cv2/__init__.py", line 8, in <module>
+            from .cv2 import *
+        ImportError: libGL.so.1: cannot open shared object file: No such file or directory
+        ```
+    - Instead, using `imageio` make gif file as output
+
+
 # Configuration
 
 `title`: _string_  

gradio_painttransformer.py

@@ -6,10 +6,8 @@
 '''
 
 import os
-os.system('sudo apt-get update')
-os.system('sudo apt-get -y install libgl1-mesa-glx')
-
-import cv2
+# import cv2  # <== error
+import imageio
 import network
 from time import time
 from glob import glob
@@ -64,23 +62,25 @@ def predict(image_file):
     final_result_list = render_serial.render_serial(original_img, net_g, meta_brushes)
     logger.info(f'--- inference took {time() - t0:.4f} sec')
 
-    out = cv2.VideoWriter('output.mp4', cv2.VideoWriter_fourcc(*'mp4v'), FPS, 
-                          (WIDTH, HEIGHT))
+    # out = cv2.VideoWriter('output.mp4', cv2.VideoWriter_fourcc(*'mp4v'), FPS, (WIDTH, HEIGHT))
+    frames = []
     for idx, frame in enumerate(final_result_list):
-        out.write(frame)
-    out.release()
+        # out.write(frame)
+        frames.append(frame)
+    # out.release()
+    imageio.mimsave('output.gif', frames, mode="RGBA")
     logger.info('--- animation generated')
-    return 'output.mp4'
+    return 'output.gif'
 
 iface = gr.Interface(
     predict,
     title='🎨 Paint Transformer',
-    description='This demo converts an image into a sequence of painted images (animation)',
+    description='This demo converts an image into a sequence of painted images (takes about 90 sec)',
     inputs=[
         gr.inputs.Image(label='Input image', type='filepath')
     ],
     outputs=[
-        gr.outputs.Video(label='Output animation', type='mp4')
+        gr.outputs.Image(label='Output animation')
     ],
     examples=examples,
     article='<p style="text-align:center">Original work: <a href="https://github.com/wzmsltw/PaintTransformer">PaintTransformer</a></p>'

requirements.txt

@@ -5,4 +5,5 @@ torch
 gradio
 Pillow
 opencv-python
+imageio
 paddlepaddle==2.2.1