Use imgviz.io.pyplot_to_numpy in examples

wkentaro · Jan 21, 2020 · 5849b86 · 5849b86
1 parent 186bc11
commit 5849b86
Show file tree

Hide file tree

Showing 19 changed files with 19 additions and 21 deletions.
diff --git a/examples/.readme/centerize.jpg b/examples/.readme/centerize.jpg
diff --git a/examples/.readme/depth2rgb.jpg b/examples/.readme/depth2rgb.jpg
diff --git a/examples/.readme/draw.jpg b/examples/.readme/draw.jpg
diff --git a/examples/.readme/flow2rgb.jpg b/examples/.readme/flow2rgb.jpg
diff --git a/examples/.readme/instances2rgb.jpg b/examples/.readme/instances2rgb.jpg
diff --git a/examples/.readme/label2rgb.jpg b/examples/.readme/label2rgb.jpg
diff --git a/examples/.readme/nchannel2rgb.jpg b/examples/.readme/nchannel2rgb.jpg
diff --git a/examples/.readme/plot_trajectory.jpg b/examples/.readme/plot_trajectory.jpg
diff --git a/examples/.readme/resize.jpg b/examples/.readme/resize.jpg
diff --git a/examples/.readme/tile.jpg b/examples/.readme/tile.jpg
diff --git a/examples/centerize.py b/examples/centerize.py
@@ -16,7 +16,7 @@ def centerize():
 
     # -------------------------------------------------------------------------
 
-    fig = plt.figure(dpi=200)
+    plt.figure(dpi=200)
 
     plt.subplot(131)
     plt.title('original')
@@ -33,10 +33,7 @@ def centerize():
     plt.imshow(centerized2)
     plt.axis('off')
 
-    img = imgviz.io.pyplot_fig2arr(fig)
-    plt.close()
-
-    return img
+    return imgviz.io.pyplot_to_numpy()
 
 
 if __name__ == '__main__':

diff --git a/examples/depth2rgb.py b/examples/depth2rgb.py
@@ -10,7 +10,7 @@ def depth2rgb():
 
     # -------------------------------------------------------------------------
 
-    fig = plt.figure(dpi=200)
+    plt.figure(dpi=200)
 
     plt.subplot(121)
     plt.title('rgb')
@@ -22,7 +22,7 @@ def depth2rgb():
     plt.imshow(depthviz)
     plt.axis('off')
 
-    img = imgviz.io.pyplot_fig2arr(fig)
+    img = imgviz.io.pyplot_to_numpy()
     plt.close()
 
     return img

diff --git a/examples/draw.py b/examples/draw.py
@@ -51,7 +51,7 @@ def draw():
 
     # -------------------------------------------------------------------------
 
-    fig = plt.figure(dpi=200)
+    plt.figure(dpi=200)
 
     plt.subplot(121)
     plt.title('original')
@@ -63,7 +63,7 @@ def draw():
     plt.imshow(viz)
     plt.axis('off')
 
-    img = imgviz.io.pyplot_fig2arr(fig)
+    img = imgviz.io.pyplot_to_numpy()
     plt.close()
 
     return img

diff --git a/examples/flow2rgb.py b/examples/flow2rgb.py
@@ -11,7 +11,8 @@ def flow2rgb():
 
     # -------------------------------------------------------------------------
 
-    fig = plt.figure(dpi=200)
+    plt.figure(dpi=200)
+
     plt.subplot(121)
     plt.title('image')
     plt.imshow(rgb)
@@ -22,7 +23,7 @@ def flow2rgb():
     plt.imshow(flowviz)
     plt.axis('off')
 
-    img = imgviz.io.pyplot_fig2arr(fig)
+    img = imgviz.io.pyplot_to_numpy()
     plt.close()
 
     return img

diff --git a/examples/instances2rgb.py b/examples/instances2rgb.py
@@ -22,7 +22,7 @@ def instances2rgb():
 
     # -------------------------------------------------------------------------
 
-    fig = plt.figure(dpi=200)
+    plt.figure(dpi=200)
 
     plt.subplot(131)
     plt.title('rgb')
@@ -39,7 +39,7 @@ def instances2rgb():
     plt.imshow(insviz2)
     plt.axis('off')
 
-    img = imgviz.io.pyplot_fig2arr(fig)
+    img = imgviz.io.pyplot_to_numpy()
     plt.close()
 
     return img

diff --git a/examples/label2rgb.py b/examples/label2rgb.py
@@ -23,7 +23,7 @@ def label2rgb():
 
     # -------------------------------------------------------------------------
 
-    fig = plt.figure(dpi=200)
+    plt.figure(dpi=200)
 
     plt.subplot(131)
     plt.title('+img')
@@ -40,7 +40,7 @@ def label2rgb():
     plt.imshow(labelviz_withname2)
     plt.axis('off')
 
-    img = imgviz.io.pyplot_fig2arr(fig)
+    img = imgviz.io.pyplot_to_numpy()
     plt.close()
 
     return img

diff --git a/examples/nchannel2rgb.py b/examples/nchannel2rgb.py
@@ -16,7 +16,7 @@ def nchannel2rgb():
 
     # -------------------------------------------------------------------------
 
-    fig = plt.figure(dpi=200)
+    plt.figure(dpi=200)
 
     plt.subplot(121)
     plt.title('rgb')
@@ -28,7 +28,7 @@ def nchannel2rgb():
     plt.imshow(nchannel_viz)
     plt.axis('off')
 
-    img = imgviz.io.pyplot_fig2arr(fig)
+    img = imgviz.io.pyplot_to_numpy()
     plt.close()
 
     return img

diff --git a/examples/resize.py b/examples/resize.py
@@ -13,7 +13,7 @@ def resize():
 
     # -------------------------------------------------------------------------
 
-    fig = plt.figure(dpi=200)
+    plt.figure(dpi=200)
 
     plt.subplot(121)
     plt.title('rgb:\n{}'.format(rgb.shape))
@@ -25,7 +25,7 @@ def resize():
     plt.imshow(rgb_resized)
     plt.axis('off')
 
-    img = imgviz.io.pyplot_fig2arr(fig)
+    img = imgviz.io.pyplot_to_numpy()
     plt.close()
 
     return img

diff --git a/examples/tile.py b/examples/tile.py
@@ -19,7 +19,7 @@ def tile():
 
     # -------------------------------------------------------------------------
 
-    fig = plt.figure(dpi=200)
+    plt.figure(dpi=200)
 
     plt.subplot(121)
     plt.title('original')
@@ -31,7 +31,7 @@ def tile():
     plt.imshow(tiled)
     plt.axis('off')
 
-    img = imgviz.io.pyplot_fig2arr(fig)
+    img = imgviz.io.pyplot_to_numpy()
     plt.close()
 
     return img