diff --git a/README.md b/README.md index 7b1d53b..0fad19b 100644 --- a/README.md +++ b/README.md @@ -75,10 +75,15 @@ cap = ffmpegcv.VideoCaptureCAM("Integrated Camera") Deeplearning pipeline. ```python -# video -> crop -> resize -> RGB -> CUDA:CHW float32 -> model +""" + —————————— NVIDIA GPU accelerating ⤴⤴ ——————— + | | + V V +video -> decode -> crop -> resize -> RGB -> CUDA:CHW float32 -> model +""" cap = ffmpegcv.toCUDA( ffmpegcv.VideoCaptureNV(file, pix_fmt='nv12', resize=(W,H)), - tensor_format='CHW') + tensor_format='chw') for frame_CHW_cuda in cap: frame_CHW_cuda = (frame_CHW_cuda - mean) / std diff --git a/README_CN.md b/README_CN.md index 7bb3612..9c7fcd9 100644 --- a/README_CN.md +++ b/README_CN.md @@ -72,10 +72,15 @@ cap = ffmpegcv.VideoCaptureCAM("Integrated Camera") 深度学习流水线 ```python -# video -> crop -> resize -> RGB -> CUDA:CHW float32 -> model +""" + —————————— NVIDIA GPU 加速 ⤴⤴ ——————— + | | + V V +视频 -> 解码器 -> 裁剪 -> 缩放 -> RGB -> CUDA:CHW float32 -> 模型 +""" cap = ffmpegcv.toCUDA( ffmpegcv.VideoCaptureNV(file, pix_fmt='nv12', resize=(W,H)), - tensor_format='CHW') + tensor_format='chw') for frame_CHW_cuda in cap: frame_CHW_cuda = (frame_CHW_cuda - mean) / std diff --git a/ffmpegcv/version.py b/ffmpegcv/version.py index d26db81..fa45835 100644 --- a/ffmpegcv/version.py +++ b/ffmpegcv/version.py @@ -1 +1 @@ -__version__='0.3.13a' \ No newline at end of file +__version__='0.3.14' \ No newline at end of file diff --git a/setup.py b/setup.py index b940f64..0ca3108 100644 --- a/setup.py +++ b/setup.py @@ -6,7 +6,7 @@ setup( name='ffmpegcv', # 应用名 - version='0.3.13a', # 版本号 + version='0.3.14', # 版本号 packages=find_packages(include=['ffmpegcv*']), # 包括在安装包内的 Python 包 author='chenxf', author_email='cxf529125853@163.com',