Browse Source
update
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
main
1 changed files with
2 additions and
2 deletions
-
video_decoder.py
|
|
@ -3,12 +3,12 @@ import av |
|
|
import numpy as np |
|
|
import numpy as np |
|
|
from towhee.operator.base import PyOperator |
|
|
from towhee.operator.base import PyOperator |
|
|
|
|
|
|
|
|
from cpu_decode import PyAVDecode |
|
|
|
|
|
|
|
|
from .cpu_decode import PyAVDecode |
|
|
|
|
|
|
|
|
logger = logging.getLogger() |
|
|
logger = logging.getLogger() |
|
|
|
|
|
|
|
|
try: |
|
|
try: |
|
|
from gpu_decode import VPFDecode |
|
|
|
|
|
|
|
|
from .gpu_decode import VPFDecode |
|
|
except Exception: |
|
|
except Exception: |
|
|
logger.error('Import GPUDecoder failed, use CPU decode') |
|
|
logger.error('Import GPUDecoder failed, use CPU decode') |
|
|
VPFDecode = PyAVDecode |
|
|
VPFDecode = PyAVDecode |
|
|
|
