mirror of
https://github.com/jantic/DeOldify.git
synced 2026-09-17 17:19:24 +08:00
Adding embedding of video
This commit is contained in:
@@ -138,6 +138,8 @@
|
||||
"import fastai\n",
|
||||
"from fasterai.visualize import *\n",
|
||||
"from pathlib import Path\n",
|
||||
"import base64\n",
|
||||
"from IPython import display as ipythondisplay\n",
|
||||
"torch.backends.cudnn.benchmark=True"
|
||||
]
|
||||
},
|
||||
@@ -186,6 +188,21 @@
|
||||
"colorizer = get_video_colorizer()"
|
||||
]
|
||||
},
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"def show_video():\n",
|
||||
" video = io.open(dest_path, 'r+b').read()\n",
|
||||
" encoded = base64.b64encode(video)\n",
|
||||
" ipythondisplay.display(HTML(data='''<video alt=\"test\" autoplay \n",
|
||||
" loop controls style=\"height: 400px;\">\n",
|
||||
" <source src=\"data:video/mp4;base64,{0}\" type=\"video/mp4\" />\n",
|
||||
" </video>'''.format(encoded.decode('ascii'))))"
|
||||
]
|
||||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"metadata": {
|
||||
@@ -220,6 +237,7 @@
|
||||
"\n",
|
||||
"if source_url is not None and source_url !='':\n",
|
||||
" colorizer.colorize_from_url(source_url, file_name, render_factor)\n",
|
||||
" show_video()\n",
|
||||
"else:\n",
|
||||
" print('Provide a video url and try again.')"
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user