Skip to content
Permalink
Browse files

fix(preprocessor-video): move sklearn dep to apply

  • Loading branch information...
hanxiao committed Jul 12, 2019
1 parent 2936da7 commit 37155bbab0123d5e8071d82f66dbfb3b816d2413
Showing with 1 addition and 1 deletion.
  1. +1 −1 gnes/preprocessor/video/shotdetect.py
@@ -16,7 +16,6 @@
# pylint: disable=low-comment-ratio

import numpy as np
from sklearn.cluster import KMeans
from .base import BaseVideoPreprocessor
from ...proto import gnes_pb2, array2blob
from ..helper import get_video_frames, compute_descriptor, compare_descriptor
@@ -39,6 +38,7 @@ def __init__(self,

def apply(self, doc: 'gnes_pb2.Document') -> None:
super().apply(doc)
from sklearn.cluster import KMeans

if doc.raw_bytes:
# stream_data = io.BytesIO(doc.raw_bytes)

0 comments on commit 37155bb

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