This repository has been archived by the owner on Oct 19, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 2.6k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Simplifying video colab by removing artistic toggle, and clarifying t…
…he render factor text.
- Loading branch information
Showing
5 changed files
with
364 additions
and
47 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,173 @@ | ||
| { | ||
| "cells": [ | ||
| { | ||
| "cell_type": "code", | ||
| "execution_count": 1, | ||
| "metadata": {}, | ||
| "outputs": [], | ||
| "source": [ | ||
| "import os\n", | ||
| "os.environ['CUDA_VISIBLE_DEVICES']='0' " | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "code", | ||
| "execution_count": 2, | ||
| "metadata": {}, | ||
| "outputs": [], | ||
| "source": [ | ||
| "from fasterai.visualize import *\n", | ||
| "plt.style.use('dark_background')\n", | ||
| "torch.backends.cudnn.benchmark=True" | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "code", | ||
| "execution_count": 3, | ||
| "metadata": {}, | ||
| "outputs": [], | ||
| "source": [ | ||
| "#Adjust render_factor (int) if image doesn't look quite right (max 64 on 11GB GPU). The default here works for most photos. \n", | ||
| "#It literally just is a number multiplied by 16 to get the square render resolution. \n", | ||
| "#Note that this doesn't affect the resolution of the final output- the output is the same resolution as the input.\n", | ||
| "#Example: render_factor=21 => color is rendered at 16x21 = 336x336 px. \n", | ||
| "render_factor=28\n", | ||
| "#Specify media_url. Many sources will work (YouTube, Imgur, Twitter, Reddit, etc). \n", | ||
| "#Complete list here: https://rg3.github.io/youtube-dl/supportedsites.html . \n", | ||
| "#NOTE: Make source_url None to just read from file at ./video/source/[file_name] directly without modification\n", | ||
| "#source_url = 'https://vimeo.com/87890004'\n", | ||
| "#source_url = 'https://www.youtube.com/watch?v=gZShc8oshtU'\n", | ||
| "#source_url = 'https://www.youtube.com/watch?v=fk6qiJjEEBo'\n", | ||
| "#source_url = 'https://twitter.com/silentmoviegifs/status/1088830101863759872'\n", | ||
| "#source_url = 'https://i.imgur.com/Ob9pZad.gifv'\n", | ||
| "source_url = None\n", | ||
| "#file_name = 'BusterKeatonTrainHouse_Original.mp4'\n", | ||
| "#file_name = 'BicycleThief_Original.mp4'\n", | ||
| "#file_name = 'ChaplinParty_Original.mp4'\n", | ||
| "file_name = 'video15.mp4'" | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "code", | ||
| "execution_count": 4, | ||
| "metadata": {}, | ||
| "outputs": [ | ||
| { | ||
| "name": "stderr", | ||
| "output_type": "stream", | ||
| "text": [ | ||
| "/media/jason/Projects/Deep Learning/DeOldifyV2/DeOldify/fastai/data_block.py:426: UserWarning: Your training set is empty. If this is by design, pass `ignore_empty=True` to remove this warning.\n", | ||
| " warn(\"Your training set is empty. If this is by design, pass `ignore_empty=True` to remove this warning.\")\n", | ||
| "/media/jason/Projects/Deep Learning/DeOldifyV2/DeOldify/fastai/data_block.py:429: UserWarning: Your validation set is empty. If this is by design, use `no_split()`\n", | ||
| " or pass `ignore_empty=True` when labelling to remove this warning.\n", | ||
| " or pass `ignore_empty=True` when labelling to remove this warning.\"\"\")\n" | ||
| ] | ||
| } | ||
| ], | ||
| "source": [ | ||
| "colorizer = get_video_colorizer(render_factor=render_factor)" | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "code", | ||
| "execution_count": 5, | ||
| "metadata": {}, | ||
| "outputs": [ | ||
| { | ||
| "data": { | ||
| "text/html": [ | ||
| "\n", | ||
| " <div>\n", | ||
| " <style>\n", | ||
| " /* Turns off some styling */\n", | ||
| " progress {\n", | ||
| " /* gets rid of default border in Firefox and Opera. */\n", | ||
| " border: none;\n", | ||
| " /* Needs to be in here for Safari polyfill so background images work as expected. */\n", | ||
| " background-size: auto;\n", | ||
| " }\n", | ||
| " .progress-bar-interrupted, .progress-bar-interrupted::-webkit-progress-bar {\n", | ||
| " background: #F44336;\n", | ||
| " }\n", | ||
| " </style>\n", | ||
| " <progress value='5172' class='' max='5172', style='width:300px; height:20px; vertical-align: middle;'></progress>\n", | ||
| " 100.00% [5172/5172 20:29<00:00]\n", | ||
| " </div>\n", | ||
| " " | ||
| ], | ||
| "text/plain": [ | ||
| "<IPython.core.display.HTML object>" | ||
| ] | ||
| }, | ||
| "metadata": {}, | ||
| "output_type": "display_data" | ||
| }, | ||
| { | ||
| "name": "stdout", | ||
| "output_type": "stream", | ||
| "text": [ | ||
| "Video created here: video/result/video15.mp4\n" | ||
| ] | ||
| } | ||
| ], | ||
| "source": [ | ||
| "if source_url is not None:\n", | ||
| " colorizer.colorize_from_url(source_url, file_name)\n", | ||
| "else:\n", | ||
| " colorizer.colorize_from_file_name(file_name)" | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "code", | ||
| "execution_count": null, | ||
| "metadata": {}, | ||
| "outputs": [], | ||
| "source": [] | ||
| } | ||
| ], | ||
| "metadata": { | ||
| "kernelspec": { | ||
| "display_name": "Python 3", | ||
| "language": "python", | ||
| "name": "python3" | ||
| }, | ||
| "language_info": { | ||
| "codemirror_mode": { | ||
| "name": "ipython", | ||
| "version": 3 | ||
| }, | ||
| "file_extension": ".py", | ||
| "mimetype": "text/x-python", | ||
| "name": "python", | ||
| "nbconvert_exporter": "python", | ||
| "pygments_lexer": "ipython3", | ||
| "version": "3.7.0" | ||
| }, | ||
| "toc": { | ||
| "colors": { | ||
| "hover_highlight": "#DAA520", | ||
| "navigate_num": "#000000", | ||
| "navigate_text": "#333333", | ||
| "running_highlight": "#FF0000", | ||
| "selected_highlight": "#FFD700", | ||
| "sidebar_border": "#EEEEEE", | ||
| "wrapper_background": "#FFFFFF" | ||
| }, | ||
| "moveMenuLeft": true, | ||
| "nav_menu": { | ||
| "height": "67px", | ||
| "width": "252px" | ||
| }, | ||
| "navigate_menu": true, | ||
| "number_sections": true, | ||
| "sideBar": true, | ||
| "threshold": 4, | ||
| "toc_cell": false, | ||
| "toc_section_display": "block", | ||
| "toc_window_display": false, | ||
| "widenNotebook": false | ||
| } | ||
| }, | ||
| "nbformat": 4, | ||
| "nbformat_minor": 2 | ||
| } |
Oops, something went wrong.