-
Notifications
You must be signed in to change notification settings - Fork 1
/
config.py
32 lines (27 loc) · 1.42 KB
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
class Config:
ROOT_PATH = '.'
# defines the maximum height dimension in pixels. Used for down-sampling the video frames
FRAME_HEIGHT = 360
CLEAR_INPUT_FRAME_CACHE = True
# defines the rate at which you want to capture frames from the input video
INPUT_FPS = 20
INPUT_VIDEO_NAME = 'input_video.mp4'
INPUT_VIDEO_PATH = f'{ROOT_PATH}/data/{INPUT_VIDEO_NAME}'
INPUT_FRAME_DIRECTORY = f'{ROOT_PATH}/data/input_frames'
INPUT_FRAME_FILE = '{:0>4d}_frame.png'
INPUT_FRAME_PATH = f'{INPUT_FRAME_DIRECTORY}/{INPUT_FRAME_FILE}'
STYLE_REF_DIRECTORY = f'{ROOT_PATH}/data/style_ref'
# defines the reference style image transition sequence. Values correspond to indices in STYLE_REF_DIRECTORY
# add None in the sequence to NOT apply style transfer for part of the video (ie. [None, 0, 1, 2])
STYLE_SEQUENCE = [0, 1]
OUTPUT_FPS = 20
OUTPUT_VIDEO_NAME = 'output_video.mp4'
OUTPUT_VIDEO_PATH = f'{ROOT_PATH}/data/{OUTPUT_VIDEO_NAME}'
OUTPUT_FRAME_DIRECTORY = f'{ROOT_PATH}/data/output_frames'
OUTPUT_FRAME_FILE = '{:0>4d}_frame.png'
OUTPUT_FRAME_PATH = f'{OUTPUT_FRAME_DIRECTORY}/{OUTPUT_FRAME_FILE}'
ARCHIVE_DIRECTORY = f'{ROOT_PATH}/data/archive'
GHOST_FRAME_TRANSPARENCY = 0.1
PRESERVE_COLORS = False
TENSORFLOW_CACHE_DIRECTORY = f'{ROOT_PATH}/tensorflow_cache'
TENSORFLOW_HUB_HANDLE = 'https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2'