opencv系列之ubuntu系统下编译python版本的opencv(指定特定的ffmpeg)

Posted 2022-02-27 狂奔的CD

正文

前置

确定安装好了cmake，nvidia-driver，Cuda，cudnn
在ubuntu中编译好对应的ffmpeg
准备将python版本的opencv，编译到conda的环境中，提前创建好自己的conda虚拟环境，我这里是cd_test, 也可以编译到其他的python环境中，具体路径得配置正确。

下载

下载opencv4.5.4
官方源码地址

下载opencv-contribute4.5.4
可以在此下载国内镜像

解压到同一目录下，结构为：
folder
|–opencv-4.5.4
|–opencv_contrib-4.5.4

编译

mkdir build
cd build
#查看cmake选项
cmake -LH ..
# 配置cmake选项，带GPU
cmake -D CMAKE_BUILD_TYPE=RELEASE \\
	-D CMAKE_INSTALL_PREFIX=../linux_install \\
	-D INSTALL_C_EXAMPLES=OFF \\
	-D INSTALL_PYTHON_EXAMPLES=OFF \\
	-D OPENCV_ENABLE_NONFREE=OFF \\
	-D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib-4.5.4/modules \\
	-D WITH_FFMPEG=on \\
	-D WITH_CUDA=ON \\
    -D WITH_CUBLAS=ON \\
    -D CUDA_FAST_MATH=ON \\
    -D WITH_CUFFT=ON \\
    -D WITH_NVCUVID=ON \\
    -D WITH_V4L=ON \\
    -D WITH_LIBV4L=ON \\
    -D WITH_OPENGL=ON \\
	-D BUILD_EXAMPLES=OFF \\
    -D OPENCV_GENERATE_PKGCONFIG=YES \\
    -D BUILD_opencv_world=NO \\
    -D BUILD_opencv_python2=OFF \\
    -D BUILD_opencv_python3=ON \\
    -D PYTHON_DEFAULT_EXECUTABLE=/home/hw/anaconda3/envs/cd_test/bin/python3 \\
    -D PYTHON3_EXECUTABLE=/home/hw/anaconda3/envs/cd_test/bin/python3 \\
    -D PYTHON3_LIBRARY=/home/hw/anaconda3/envs/cd_test/lib/libpython3.8.so \\
    -D PYTHON_INCLUDE_DIR=/home/hw/anaconda3/envs/cd_test/include/python3.8 \\
    -D PYTHON3_NUMPY_INCLUDE_DIRS=/home/hw/anaconda3/envs/cd_test/lib/python3.8/site-packages/numpy/core/include \\
    -D PYTHON3_PACKAGES_PATH=/home/hw/anaconda3/envs/cd_test/lib/python3.8/site-packages \\
    ..

# ffmpeg版本
cmake -D CMAKE_BUILD_TYPE=RELEASE \\
	-D CMAKE_INSTALL_PREFIX=../linux_install \\
	-D INSTALL_C_EXAMPLES=OFF \\
	-D INSTALL_PYTHON_EXAMPLES=OFF \\
	-D OPENCV_ENABLE_NONFREE=OFF \\
	-D WITH_FFMPEG=ON \\
	-D WITH_GSTREAMER=OFF \\
	-D BUILD_EXAMPLES=OFF \\
	-D OPENCV_GENERATE_PKGCONFIG=YES \\
    -D BUILD_opencv_world=OFF \\
    -D BUILD_opencv_python2=OFF \\
    -D BUILD_opencv_python3=ON \\
    -D PYTHON_DEFAULT_EXECUTABLE=/home/hw/anaconda3/envs/cd_test/bin/python3 \\
    -D PYTHON3_EXECUTABLE=/home/hw/anaconda3/envs/cd_test/bin/python3 \\
    -D PYTHON3_LIBRARY=/home/hw/anaconda3/envs/cd_test/lib/libpython3.8.so \\
    -D PYTHON_INCLUDE_DIR=/home/hw/anaconda3/envs/cd_test/include/python3.8 \\
    -D PYTHON3_NUMPY_INCLUDE_DIRS=/home/hw/anaconda3/envs/cd_test/lib/python3.8/site-packages/numpy/core/include \\
    -D PYTHON3_PACKAGES_PATH=/home/hw/anaconda3/envs/cd_test/lib/python3.8/site-packages \\
    ..

说明：如果该命令中不加-D CMAKE_INSTALL_PREFIX,则默认各部分分别安装在/usr/local/目录的include/, bin/, lib/3个文件夹下。
-D OPENCV_GENERATE_PKGCONFIG=YES: OpenCV4以上版本默认不使用pkg-config，该编译选项开启生成opencv4.pc文件，支持pkg-config功能。
-D WITH_CUDA=ON实现和cuda的联合编译。
后面关于python的路径设置以自己机子anaconda的设定为准，需要注意的是python版本和你使用的虚拟环境一致。

【问题1】fatal error: ImfChromaticities.h: No such file or directory
【解决办法】 sudo apt-get install libopenexr-dev

【问题2】ffmpeg选项是no

-- OpenCV Python: during development append to PYTHONPATH: /dataset/users/chengd/envs/opencv/opencv-4.5.4/build/python_loader
-- WARNING: Can't build ffmpeg test code

【解决办法】首先记得删除build文件夹中的缓存文件，重新cmake,不然会还是no
其次cmake过程会打印no的原因，比如我这里是so没找着

最后发现编译生成的/usr/local/ffmpeg3.4.9/lib/pkgconfig/libavcodec.pc中的路径是错的,
给改到正确位置

同时记得删除build文件夹中的缓存文件，不然会还是no。
解决完问题，就可以顺利找到了

【问题3】编译时ffmpeg的某些函数找不着
【解决办法】最后发现时ffmpeg的版本问题，ffmpeg最新版很多方法更新移除了，目前opencv4.x版本编译与ffmpeg3.x版本吻合。所以这里编译了ffmpg3.4.9版本

安装

make -j6
sudo make install

运行

conda activate cd_test
#此时在conda和pip的包管理器中都没有opencv-python,但是可以运行
python
>>> import cv2
>>> print(cv2.__version__)
4.5.4

可能提示你先要安装numpy，装一下就好了pip install numpy