polish to CUDA feature

chenxinfeng4 · Jun 13, 2024 · 239710c · 239710c
1 parent b0768f4
commit 239710c
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 6 deletions.
diff --git 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
@@ -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
@@ -1 +1 @@
-__version__='0.3.13a'
+__version__='0.3.14'
diff --git 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='[email protected]',