diff --git a/README.md b/README.md index 27cbefb..aa2c829 100644 --- a/README.md +++ b/README.md @@ -48,6 +48,7 @@ Options: ``` ## Example +**Single file** ```shell $ ./generator videos/27467_1_milkbots_wg_720p.mp4 2 126 73 10 thumbnails.jpg Extracting 5 frames @@ -56,6 +57,15 @@ Frames extracted. Saved! ``` +**Directory** +```shell +$ ./generator videos/ 2 126 73 10 thumbnails/ +Extracting 5 frames + [####################################] 100% +Frames extracted. +Saved! +``` + ![image](https://cloud.githubusercontent.com/assets/244265/11234316/b42913a6-8d94-11e5-865a-128ea8d801f7.png) diff --git a/generator b/generator index 60cea15..164862c 100755 --- a/generator +++ b/generator @@ -22,28 +22,63 @@ from docopt import docopt from moviepy.editor import VideoFileClip from PIL import Image from click import progressbar +from collections import namedtuple import glob import os import random import shutil import math import tempfile +import sys TMP_FRAMES_PATH = tempfile.mkdtemp() def generate_video_thumbnails(args): - filepath = args['