Skip to content
This repository has been archived by the owner on Feb 22, 2020. It is now read-only.

Commit

Permalink
fix(encoder): fix import path for mixture encoder
Browse files Browse the repository at this point in the history
  • Loading branch information
Larryjianfeng committed Aug 1, 2019
1 parent 1777967 commit d6a46fa
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions gnes/encoder/video/incep_mixture.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -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())

Expand Down

0 comments on commit d6a46fa

Please sign in to comment.