Skip to content
Permalink
Browse files

fix(encoder): fix import path for mixture encoder

  • Loading branch information...
Larryjianfeng committed Aug 1, 2019
1 parent 1777967 commit d6a46fa6b6a931b4dc0e334f5a89bb66460a84b0
Showing with 6 additions and 6 deletions.
  1. +6 −6 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())

0 comments on commit d6a46fa

Please sign in to comment.
You can’t perform that action at this time.