Преглед на файлове

Simplifying colab notebook

Jason Antic преди 6 години
родител
ревизия
ee61145c17
променени са 1 файла, в които са добавени 1 реда и са изтрити 16 реда
  1. 1 16
      DeOldify_colab.ipynb

+ 1 - 16
DeOldify_colab.ipynb

@@ -243,22 +243,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "data = get_dummy_databunch()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "learn = colorize_gen_learner(data=data, nf_factor=1.25)\n",
-    "#switch to read models from proper place\n",
-    "learn.path = Path('./')\n",
-    "learn.load(weights_name)\n",
-    "learn.model.eval()\n",
-    "filtr = MasterFilter([ColorizerFilter(learn=learn)], render_factor=render_factor)\n",
-    "vis = ModelImageVisualizer(filtr, results_dir=results_dir)"
+    "vis = get_colorize_visualizer(results_dir=results_dir, render_factor=render_factor)"
    ]
   },
   {