From 7927412f796e6e72cfbb47b789e30a3cf61b9f5b Mon Sep 17 00:00:00 2001 From: N00MKRAD Date: Sat, 13 Mar 2021 17:16:34 +0100 Subject: [PATCH] flavr.py: Set working dir before trying to import dataset This enables compatibility with embedded python. --- Pkgs/flavr-cuda/flavr.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/Pkgs/flavr-cuda/flavr.py b/Pkgs/flavr-cuda/flavr.py index 193cb87..1497ad6 100644 --- a/Pkgs/flavr-cuda/flavr.py +++ b/Pkgs/flavr-cuda/flavr.py @@ -9,8 +9,6 @@ from PIL import Image import numpy as np import _thread from torchvision.io import read_video, write_video -from dataset.transforms import ToTensorVideo, Resize - import torch.nn.functional as F abspath = os.path.abspath(__file__) @@ -20,6 +18,8 @@ os.chdir(os.path.dirname(dname)) print("Added {0} to temporary PATH".format(dname)) sys.path.append(dname) +from dataset.transforms import ToTensorVideo, Resize + import argparse parser = argparse.ArgumentParser() @@ -42,8 +42,6 @@ path = args.input base = os.path.basename(path) interp_input_path = os.path.join(dname, args.input) interp_output_path = os.path.join(dname, args.output) -print("\interp_input_path: " + interp_input_path) -print("\ninterp_output_path: " + interp_output_path) if args.input.endswith("/"): video_name = args.input.split("/")[-2].split(input_ext)[0]