Explorar o código

Optimizing colabs for speed on image downloads with render previews

Jason Antic %!s(int64=6) %!d(string=hai) anos
pai
achega
1227f9e838
Modificáronse 3 ficheiros con 3 adicións e 3 borrados
  1. 1 1
      ImageColorizerColab.ipynb
  2. 1 1
      VideoColorizerColab.ipynb
  3. 1 1
      fasterai/visualize.py

+ 1 - 1
ImageColorizerColab.ipynb

@@ -221,7 +221,7 @@
    "outputs": [],
    "source": [
     "for i in range(5,46):\n",
-    "    colorizer.plot_transformed_image('test_images/image.png', render_factor=i, display_render_factor=True)"
+    "    colorizer.plot_transformed_image('test_images/image.png', render_factor=i, display_render_factor=True, figsize=(10,10))"
    ]
   },
   {

+ 1 - 1
VideoColorizerColab.ipynb

@@ -232,7 +232,7 @@
    "outputs": [],
    "source": [
     "for i in range(5,44):\n",
-    "    colorizer.vis.plot_transformed_image('video/bwframes/video/00001.jpg', render_factor=i, display_render_factor=True)"
+    "    colorizer.vis.plot_transformed_image('video/bwframes/video/00001.jpg', render_factor=i, display_render_factor=True, figsize=(10,10))"
    ]
   },
   {

+ 1 - 1
fasterai/visualize.py

@@ -70,7 +70,7 @@ class ModelImageVisualizer():
         axes.imshow(np.asarray(image)/255)
         axes.axis('off')
         if render_factor is not None and display_render_factor:
-            plt.text(10,10,'render_factor: ' + str(render_factor), backgroundcolor='black')
+            plt.text(10,10,'render_factor: ' + str(render_factor), color='white', backgroundcolor='black')
 
     def _get_num_rows_columns(self, num_images:int, max_columns:int)->(int,int):
         columns = min(num_images, max_columns)