diff --git a/scripts/combine_results.py b/scripts/combine_results.py index 2b684ac..7b672af 100644 --- a/scripts/combine_results.py +++ b/scripts/combine_results.py @@ -10,6 +10,9 @@ import cv2 import numpy as np +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(os.path.realpath(__file__)),'..'))) + from video_prediction.utils import html from video_prediction.utils.ffmpeg_gif import save_gif as ffmpeg_save_gif diff --git a/scripts/evaluate.py b/scripts/evaluate.py index 2f7f4c3..4df48e1 100644 --- a/scripts/evaluate.py +++ b/scripts/evaluate.py @@ -13,6 +13,9 @@ import numpy as np import tensorflow as tf +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(os.path.realpath(__file__)),'..'))) + from video_prediction import datasets, models diff --git a/scripts/generate.py b/scripts/generate.py index 44b4b42..41f0b8b 100644 --- a/scripts/generate.py +++ b/scripts/generate.py @@ -12,6 +12,9 @@ import numpy as np import tensorflow as tf +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(os.path.realpath(__file__)),'..'))) + from video_prediction import datasets, models from video_prediction.utils.ffmpeg_gif import save_gif diff --git a/scripts/train.py b/scripts/train.py index 78502a6..7e735de 100644 --- a/scripts/train.py +++ b/scripts/train.py @@ -12,6 +12,9 @@ import numpy as np import tensorflow as tf +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(os.path.realpath(__file__)),'..'))) + from video_prediction import datasets, models