opencv‘s dnn moudle is now support for Nvidia cuda。
ccl-private opened this issue · 1 comments
opencv‘s dnn moudle is now support for Nvidia cuda, and needs opencv version of 4.2.0. Can you updata the opencv version and build opencv with DNN_CUDA support? Please~
below is buildOpencv .sh, i hope help you
#!/bin/bash
License: MIT. See license file in root directory
Copyright(c) JetsonHacks (2017-2018)
OPENCV_VERSION=4.4.0
Jetson TX2
ARCH_BIN=6.2
Jetson TX1
ARCH_BIN=5.3
INSTALL_DIR=/usr/local
Download the opencv_extras repository
If you are installing the opencv testdata, ie
OPENCV_TEST_DATA_PATH=../opencv_extra/testdata
Make sure that you set this to YES
Value should be YES or NO
DOWNLOAD_OPENCV_EXTRAS=YES
Source code directory
OPENCV_SOURCE_DIR=$HOME
WHEREAMI=$PWD
CLEANUP=true
function usage
{
echo "usage: ./buildOpenCVTX1.sh [[-s sourcedir ] | [-h]]"
echo "-s | --sourcedir Directory in which to place the opencv sources (default $HOME)"
echo "-i | --installdir Directory in which to install opencv libraries (default /usr/local)"
echo "-h | --help This message"
}
Iterate through command line inputs
while [ "$1" != "" ]; do
case $1 in
-s | --sourcedir ) shift
OPENCV_SOURCE_DIR=$1
;;
-i | --installdir ) shift
INSTALL_DIR=$1
;;
-h | --help ) usage
exit
;;
* ) usage
exit 1
esac
shift
done
CMAKE_INSTALL_PREFIX=$INSTALL_DIR
source scripts/jetson_variables.sh
Print out the current configuration
echo "Build configuration: "
echo " NVIDIA Jetson $JETSON_BOARD"
echo " Operating System: $JETSON_L4T_STRING [Jetpack $JETSON_JETPACK]"
echo " Current OpenCV Installation: $JETSON_OPENCV"
echo " OpenCV binaries will be installed in: $CMAKE_INSTALL_PREFIX"
echo " OpenCV Source will be installed in: $OPENCV_SOURCE_DIR"
if [ $DOWNLOAD_OPENCV_EXTRAS == "YES" ] ; then
echo "Also installing opencv_extras"
fi
#improtant security updates
sudo add-apt-repository "deb http://ports.ubuntu.com/ubuntu-ports xenial-security main"
Repository setup
sudo apt-add-repository universe
sudo apt-get update
Download dependencies for the desired configuration
cd $WHEREAMI
sudo apt-get install -y
cmake
libavcodec-dev
libavformat-dev
libavutil-dev
libeigen3-dev
libglew-dev
libgtk2.0-dev
libgtk-3-dev
libpostproc-dev
libswscale-dev
libtbb-dev
libtiff5-dev
libv4l-dev
libxvidcore-dev
libx264-dev
qt5-default
zlib1g-dev
libjasper-dev
libjpeg-dev
libpng-dev
pkg-config
https://devtalk.nvidia.com/default/topic/1007290/jetson-tx2/building-opencv-with-opengl-support-/post/5141945/#5141945
cd /usr/local/cuda/include
sudo patch -N cuda_gl_interop.h $WHEREAMI'/patches/OpenGLHeader.patch'
Clean up the OpenGL tegra libs that usually get crushed
cd /usr/lib/aarch64-linux-gnu/
sudo ln -s libGL.so.1 libGL.so
Python 2.7
sudo apt-get install -y python-dev python-numpy python-py python-pytest
Python 3.5
sudo apt-get install -y python3-dev python3-numpy python3-py python3-pytest
GStreamer support
sudo apt-get install -y libgstreamer1.0-dev libgstreamer-plugins-base1.0-dev
cd $OPENCV_SOURCE_DIR
git clone https://github.com/opencv/opencv.git
cd opencv
git checkout -b v${OPENCV_VERSION} ${OPENCV_VERSION}
if [ $OPENCV_VERSION = 3.4.1 ] ; then
For 3.4.1, use this commit to fix samples/gpu/CMakeLists.txt
git merge ec0bb66
For 3.4.1, use this to fix C compilation issues (like in YOLO)
git cherry-pick 549b5df
fi
if [ $DOWNLOAD_OPENCV_EXTRAS == "YES" ] ; then
echo "Installing opencv_extras"
This is for the test data
cd $OPENCV_SOURCE_DIR
#git clone https://github.com/opencv/opencv_extra.git
#cd opencv_extra
#git checkout -b v${OPENCV_VERSION} ${OPENCV_VERSION}
git clone https://github.com/opencv/opencv_contrib.git
cd opencv_contrib
git checkout -b v${OPENCV_VERSION} ${OPENCV_VERSION}
fi
cd $OPENCV_SOURCE_DIR/opencv
mkdir build
cd build
Here are some options to install source examples and tests
-D INSTALL_TESTS=ON \
-D OPENCV_TEST_DATA_PATH=../opencv_extra/testdata \
-D INSTALL_C_EXAMPLES=ON \
-D INSTALL_PYTHON_EXAMPLES=ON \
There are also switches which tell CMAKE to build the samples and tests
Check OpenCV documentation for details
time cmake -D CMAKE_BUILD_TYPE=RELEASE
-D CMAKE_INSTALL_PREFIX=${CMAKE_INSTALL_PREFIX}
-D WITH_CUDA=ON
-D CUDA_ARCH_BIN=${ARCH_BIN}
-D CUDA_ARCH_PTX=""
-D ENABLE_FAST_MATH=ON
-D CUDA_FAST_MATH=ON
-D WITH_CUBLAS=ON
-D WITH_LIBV4L=ON
-D WITH_GSTREAMER=ON
-D WITH_GSTREAMER_0_10=OFF
-D WITH_QT=ON
-D WITH_OPENGL=ON
-D OPENCV_EXTRA_MODULES_PATH=/home/nvidia/opencv_contrib/modules
-D CUDNN_LIBRARY=/usr/local/cuda/lib64/libcudnn.so.8.0.0
-D CUDNN_VERSION='8.0'
../
if [ $? -eq 0 ] ; then
echo "CMake configuration make successful"
else
Try to make again
echo "CMake issues " >&2
echo "Please check the configuration being used"
exit 1
fi
Consider $ sudo nvpmodel -m 2 or $ sudo nvpmodel -m 0
NUM_CPU=$(nproc)
time make -j$(($NUM_CPU - 1))
if [ $? -eq 0 ] ; then
echo "OpenCV make successful"
else
Try to make again; Sometimes there are issues with the build
because of lack of resources or concurrency issues
echo "Make did not build " >&2
echo "Retrying ... "
Single thread this time
make
if [ $? -eq 0 ] ; then
echo "OpenCV make successful"
else
# Try to make again
echo "Make did not successfully build" >&2
echo "Please fix issues and retry build"
exit 1
fi
fi
echo "Installing ... "
sudo make install
if [ $? -eq 0 ] ; then
echo "OpenCV installed in: $CMAKE_INSTALL_PREFIX"
else
echo "There was an issue with the final installation"
exit 1
fi
check installation
IMPORT_CHECK="$(python -c "import cv2 ; print cv2.version")"
if [[ $IMPORT_CHECK != $OPENCV_VERSION ]]; then
echo "There was an error loading OpenCV in the Python sanity test."
echo "The loaded version does not match the version built here."
echo "Please check the installation."
echo "The first check should be the PYTHONPATH environment variable."
fi