From d6a46fa6b6a931b4dc0e334f5a89bb66460a84b0 Mon Sep 17 00:00:00 2001 From: Larry Yan Date: Thu, 1 Aug 2019 11:32:37 +0800 Subject: [PATCH] fix(encoder): fix import path for mixture encoder --- gnes/encoder/video/incep_mixture.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/gnes/encoder/video/incep_mixture.py b/gnes/encoder/video/incep_mixture.py index eac796e4..60a7710d 100644 --- a/gnes/encoder/video/incep_mixture.py +++ b/gnes/encoder/video/incep_mixture.py @@ -18,11 +18,11 @@ import numpy as np from PIL import Image -from gnes.encoder.base import BaseImageEncoder -from gnes.helper import batching, batch_iterator, get_first_available_gpu +from ..base import BaseVideoEncoder +from ...helper import batching, batch_iterator, get_first_available_gpu -class IncepMixtureEncoder(BaseImageEncoder): +class IncepMixtureEncoder(BaseVideoEncoder): def __init__(self, model_dir_inception: str, model_dir_mixture: str, @@ -57,9 +57,9 @@ def __init__(self, model_dir_inception: str, def post_init(self): import tensorflow as tf - from gnes.encoder.image.inception_cores.inception_v4 import inception_v4 - from gnes.encoder.image.inception_cores.inception_utils import inception_arg_scope - from gnes.encoder.video.mixture_core.model import NetFV + from ..image.inception_cores.inception_v4 import inception_v4 + from ..image.inception_cores.inception_utils import inception_arg_scope + from .mixture_core.model import NetFV import os os.environ['CUDA_VISIBLE_DEVICES'] = str(get_first_available_gpu())