|
@@ -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)"
|
|
|
]
|
|
|
},
|
|
|
{
|