From bd0ed247e93e7e93c825530ab28437cdfc7f7029 Mon Sep 17 00:00:00 2001
From: Micaela Verucchi <micaela.verucchi@unimore.it>
Date: Mon, 6 Apr 2020 17:27:25 +0200
Subject: [PATCH] Add opencv install script (for real)

Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
---
 install_OpenCV4.sh | 66 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100644 install_OpenCV4.sh

diff --git a/install_OpenCV4.sh b/install_OpenCV4.sh
new file mode 100644
index 0000000..8862cdc
--- /dev/null
+++ b/install_OpenCV4.sh
@@ -0,0 +1,66 @@
+#!/bin/bash
+
+#based on https://devtalk.nvidia.com/default/topic/1042035/installing-opencv4-on-xavier/ & https://github.com/markste-in/OpenCV4XAVIER/blob/master/buildOpenCV4.sh
+
+# Compute Capabilities can be found here https://developer.nvidia.com/cuda-gpus#compute
+ARCH_BIN=7.2 # AGX Xavier
+#ARCH_BIN=6.2 # Tx2
+
+cd ~/Downloads
+sudo apt-get install -y build-essential \
+    unzip \
+    pkg-config \
+    libjpeg-dev \
+    libpng-dev \
+    libtiff-dev \
+    libavcodec-dev \
+    libavformat-dev \
+    libswscale-dev \
+    libv4l-dev \
+    libxvidcore-dev \
+    libx264-dev \
+    libgtk-3-dev \
+    libatlas-base-dev \
+    gfortran \
+    python3-dev \
+    python3-venv \
+    libgstreamer1.0-dev \
+    libgstreamer-plugins-base1.0-dev \
+    libdc1394-22-dev \
+    libavresample-dev
+
+git clone https://github.com/opencv/opencv.git
+git clone https://github.com/opencv/opencv_contrib.git
+
+python3 -m venv opencv4
+source opencv4/bin/activate
+pip install wheel
+pip install numpy
+
+cd opencv && mkdir build && cd build
+
+cmake -D CMAKE_BUILD_TYPE=RELEASE \
+    -D CMAKE_INSTALL_PREFIX=/usr/local \
+    -D INSTALL_PYTHON_EXAMPLES=ON \
+    -D INSTALL_C_EXAMPLES=OFF \
+    -D OPENCV_EXTRA_MODULES_PATH='~/Downloads/opencv_contrib/modules' \
+    -D PYTHON_EXECUTABLE='~/Downloads/opencv4/bin/python' \
+    -D BUILD_EXAMPLES=ON \
+    -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_TBB=ON \
+    ../
+
+make -j4
+sudo make install
+sudo ldconfig
+
+cd '~/Downloads/opencv4/lib/python3.6/site-packages'
+ln -s /usr/local/lib/python3.6/site-packages/cv2.cpython-36m-aarch64-linux-gnu.so cv2.so