buildOpenCVTX2 icon indicating copy to clipboard operation
buildOpenCVTX2 copied to clipboard

opencv‘s dnn moudle is now support for Nvidia cuda。

Open ccl-private opened this issue 4 years ago • 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~

ccl-private avatar Mar 25 '20 04:03 ccl-private

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

majiang2018 avatar Sep 11 '20 04:09 majiang2018