/
bombuscv-raspi.sh
executable file
·190 lines (169 loc) · 6.2 KB
/
bombuscv-raspi.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
#!/usr/bin/env sh
RED="\e[1;31m"
YELLOW="\e[1;33m"
GREEN="\e[1;32m"
CYAN="\e[1;36m"
NORM="\e[0m"
echo "$YELLOW██████╗ ██████╗ ███╗ ███╗██████╗ ██╗ ██╗███████╗ ██████╗██╗ ██╗"
echo "██╔══██╗██╔═══██╗████╗ ████║██╔══██╗██║ ██║██╔════╝██╔════╝██║ ██║"
echo "$NORM██████╔╝██║ ██║██╔████╔██║██████╔╝██║ ██║███████╗██║ ██║ ██║"
echo "██╔══██╗██║ ██║██║╚██╔╝██║██╔══██╗██║ ██║╚════██║██║ ╚██╗ ██╔╝"
echo "$YELLOW██████╔╝╚██████╔╝██║ ╚═╝ ██║██████╔╝╚██████╔╝███████║╚██████╗ ╚████╔╝ "
echo "╚═════╝ ╚═════╝ ╚═╝ ╚═╝╚═════╝ ╚═════╝ ╚══════╝ ╚═════╝ ╚═══╝ $NORM"
SWAP_FILE=/etc/dphys-swapfile
greet ()
{
echo "$CYAN#######################################################"
echo "## Congratulations! BombusCV successfully installed! ##"
echo "#######################################################$NORM"
}
echo "$CYAN#####################################################################"
echo "## Installation helper script for bombuscv-rs (by Marco Radocchia) ##"
echo "## Requirement: RaspberryPi 4 (4/8GB), RaspberryPi OS aarch64 ##"
echo "## Warning: the installation process may take a while (>1h)... ##"
echo "#####################################################################$NORM"
# Check if Raspberry Pi is running RaspberryPi OS 64 bits:
[ $(uname -m) != "aarch64" -o $(command -v apt-get | wc -l) != 1 ] && \
echo "$RED==> Error:$NORM please install RaspberryPi OS 64 bits and retry." && \
exit 1
# Check if Raspberry is at least 4GB RAM.
[ $(free --mebi | grep -e "^Mem:" | awk '{print $2}') -lt 3000 ] && \
echo "$RED==> Error:$NORM required at least 4GB of RAM." && exit 1
# Update the system.
echo "$GREEN==> Updating the system...$NORM"
sudo apt-get -y update && sudo apt-get -y upgrade
# Update bootloader.
echo "$GREEN==> Updating bootloader...$NORM"
sudo rpi-eeprom-update -a
# Bring gpu memory up to 256MB.
echo "$GREEN==> Increasing gpu memory...$NORM"
sudo sed -i /boot/config.txt -e s'/gpu_mem=.*/gpu_mem=256/'
# Increasing swap size.
echo "$GREEN==> Increasing swap size...$NORM"
# storing the original swap size for later restore
orig_swap=$(awk -F'=' '/CONF_SWAPSIZE=/ {print $2}' $SWAP_FILE)
sudo sed -i $SWAP_FILE -e s'/CONF_SWAPSIZE=.*/CONF_SWAPSIZE=4096/'
sudo /etc/init.d/dphys-swapfile restart
# Install all dependencies with apt-get.
echo "$GREEN==> Installing dependencies...$NORM"
sudo apt-get install -y \
clang \
libclang-dev \
build-essential \
cmake \
git \
ffmpeg \
unzip \
pkg-config \
libjpeg-dev \
libpng-dev \
libavcodec-dev \
libavformat-dev \
libswscale-dev \
libgstreamer1.0-dev \
libgstreamer-plugins-base1.0-dev \
gstreamer1.0-gl \
libxvidcore-dev \
libx264-dev \
libtbb2 \
libtbb-dev \
libdc1394-22-dev \
libv4l-dev \
v4l-utils \
libopenblas-dev \
libatlas-base-dev \
libblas-dev \
liblapack-dev \
gfortran \
libhdf5-dev \
libprotobuf-dev \
libgoogle-glog-dev \
libgflags-dev \
protobuf-compiler
# Download OpenCV 4.6.0.
wget -O opencv.zip https://github.com/opencv/opencv/archive/4.6.0.zip
wget -O opencv_contrib.zip https://github.com/opencv/opencv_contrib/archive/4.6.0.zip
# unzip downloaded files
unzip opencv.zip
unzip opencv_contrib.zip
# rename directories for convenience
mv opencv-4.6.0 opencv
mv opencv_contrib-4.6.0 opencv_contrib
# remove the zip files
rm opencv.zip
rm opencv_contrib.zip
# create the build directory
cd opencv && mkdir build && cd build
# Compile OpenCV 4.6.0.
echo "$GREEN==> Compiling OpenCV v4.6.0...$NORM"
# run cmake
cmake -DCMAKE_BUILD_TYPE=RELEASE \
-DCMAKE_INSTALL_PREFIX=/usr/local \
-DOPENCV_EXTRA_MODULES_PATH=$HOME/opencv_contrib/modules \
-DCPU_BASELINE=NEON \
-DENABLE_NEON=ON \
-DWITH_OPENMP=ON \
-DWITH_OPENCL=OFF \
-DBUILD_TIFF=ON \
-DWITH_FFMPEG=ON \
-DWITH_TBB=ON \
-DBUILD_TBB=ON \
-DWITH_GSTREAMER=ON \
-DBUILD_TESTS=OFF \
-DWITH_EIGEN=OFF \
-DWITH_V4L=ON \
-DWITH_LIBV4L=ON \
-DWITH_VTK=OFF \
-DWITH_QT=OFF \
-DWITH_GTK=OFF \
-DHIGHGUI_ENABLE_PLUGINS=OFF \
-DWITH_WIN32UI=OFF \
-DWITH_DSHOW=OFF \
-DWITH_AVFOUNDATION=OFF \
-DWITH_MSMF=OFF \
-DWITH_TESTs=OFF \
-DOPENCV_ENABLE_NONFREE=ON \
-DINSTALL_C_EXAMPLES=OFF \
-DINSTALL_PYTHON_EXAMPLES=OFF \
-DINSTALL_ANDROID_EXAMPLES=OFF \
-DWITH_ANDROID_MEDIANDK=OFF \
-DINSTALL_BIN_EXAMPLES=OFF \
-DOPENCV_GENERATE_PKGCONFIG=ON \
-DBUILD_EXAMPLES=OFF \
-DBUILD_JAVA=OFF \
-DBUILD_FAT_JAVA_LIB=OFF \
-DBUILD_JAVA=OFF \
-DBUILD_opencv_python2=OFF \
-DBUILD_opencv_python3=OFF \
-DENABLE_PYLINT=OFF \
-DENABLE_FLAKE8=OFF \
..
# run make using all 4 cores
make -j4
# Install OpenCV 4.6.0
echo "$GREEN==> Installing OpenCV v4.6.0...$NORM"
sudo make install
sudo ldconfig
# changing cwd back to $HOME
cd $HOME
# Remove opencv source directories.
echo "$GREEN==> Removing OpenCV files...$NORM"
rm -rf $HOME/opencv
rm -rf $HOME/opencv_contrib
# Install rustup.
echo "$GREEN==> Installing rustup...$NORM"
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
# Cargo install bombuscv-rs if rustup successfully installed cargo.
if [ $? = 0 ]; then
echo "$GREEN==> Installing bombuscv-rs...$NORM"
$HOME/.cargo/bin/cargo install bombuscv-rs && greet
else
echo "$RED==> Error:$NORM unable to install rustup, please retry."
fi
# Restoring swap size
echo "$GREEN==> Restoring swap size...$NORM"
sudo sed -i $SWAP_FILE -e s"/CONF_SWAPSIZE=.*/CONF_SWAPSIZE=$orig_swap/"
sudo /etc/init.d/dphys-swapfile restart
echo "$CYAN#################################################"
echo "## Please reboot your system before continung. ##"
echo "#################################################$NORM"