mathieu/easy-local-alpr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

revert shared libraries stripping, messes with tensorflow and use tensorflow v1 instead of v2 on cpu (smaller size)

Browse Source
This commit is contained in:
Mathieu Broillet 2024-08-05 11:32:18 +02:00
parent 59a54945ff
commit 145452b0ae
Signed by: mathieu
GPG Key ID: C4A6176ABC6B2DFC
1 changed files with 19 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
build_and_setup_ultimatealvr.sh
View File

@ -1,5 +1,7 @@
#!/bin/bash #!/bin/bash
deactivate 2>/dev/null
# Function to create virtual environment, install the wheel, and copy assets and libs # Function to create virtual environment, install the wheel, and copy assets and libs
install_and_setup() { install_and_setup() {
echo "Creating virtual environment at the root..." echo "Creating virtual environment at the root..."
@ -55,7 +57,7 @@ prompt_auto_setup() {
esac esac
} }
# Directories # Variables
ROOT_DIR=$(pwd) ROOT_DIR=$(pwd)
BUILD_DIR="$ROOT_DIR/tmp-build-env" BUILD_DIR="$ROOT_DIR/tmp-build-env"
SDK_ZIP_URL="https://github.com/DoubangoTelecom/ultimateALPR-SDK/archive/8130c76140fe8edc60fe20f875796121a8d22fed.zip" SDK_ZIP_URL="https://github.com/DoubangoTelecom/ultimateALPR-SDK/archive/8130c76140fe8edc60fe20f875796121a8d22fed.zip"
@ -69,7 +71,12 @@ cd "$BUILD_DIR" || exit
# Clone SDK # Clone SDK
echo "Downloading SDK..." echo "Downloading SDK..."
if [ -f "$SDK_ZIP" ]; then
echo "SDK zip already exists."
rm -R "$SDK_DIR"
else
wget "$SDK_ZIP_URL" -O "$SDK_ZIP" >/dev/null 2>&1 wget "$SDK_ZIP_URL" -O "$SDK_ZIP" >/dev/null 2>&1
fi
if [ $? -ne 0 ]; then if [ $? -ne 0 ]; then
echo "Failed to download SDK." echo "Failed to download SDK."
exit 1 exit 1
@ -121,26 +128,30 @@ read -r -p "Do you want TensorFlow for CPU or GPU? (cpu/gpu): " tf_choice
mkdir -p "$BIN_DIR/tensorflow" mkdir -p "$BIN_DIR/tensorflow"
if [ "$tf_choice" == "gpu" ]; then if [ "$tf_choice" == "gpu" ]; then
echo "Downloading TensorFlow GPU..." echo "Downloading TensorFlow GPU..."
wget https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-2.6.0.tar.gz >/dev/null 2>&1 wget https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-2.6.0.tar.gz >/dev/null 2>&1 # Use 2.6 for newer GPU support
if [ $? -ne 0 ]; then if [ $? -ne 0 ]; then
echo "Failed to download TensorFlow GPU." echo "Failed to download TensorFlow GPU."
exit 1 exit 1
fi fi
echo "Extracting TensorFlow GPU..." echo "Extracting TensorFlow GPU..."
tar -xf libtensorflow-gpu-linux-x86_64-2.6.0.tar.gz -C "$BIN_DIR/tensorflow" >/dev/null 2>&1 tar -xf libtensorflow-gpu-linux-x86_64-2.6.0.tar.gz -C "$BIN_DIR/tensorflow" >/dev/null 2>&1
mv "$BIN_DIR/tensorflow/lib/libtensorflow.so.1" "$BIN_DIR/libs/libtensorflow.so.1"
mv "$BIN_DIR/tensorflow/lib/libtensorflow_framework.so.2.6.0" "$BIN_DIR/libs/libtensorflow_framework.so.2"
else else
echo "Downloading TensorFlow CPU..." echo "Downloading TensorFlow CPU..."
wget https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.6.0.tar.gz >/dev/null 2>&1 #wget https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.6.0.tar.gz >/dev/null 2>&1
wget https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-1.14.0.tar.gz >/dev/null 2>&1 # Use 1.14 as it's smaller in size
if [ $? -ne 0 ]; then if [ $? -ne 0 ]; then
echo "Failed to download TensorFlow CPU." echo "Failed to download TensorFlow CPU."
exit 1 exit 1
fi fi
echo "Extracting TensorFlow CPU..." echo "Extracting TensorFlow CPU..."
tar -xf libtensorflow-cpu-linux-x86_64-2.6.0.tar.gz -C "$BIN_DIR/tensorflow" >/dev/null 2>&1 tar -xf libtensorflow-cpu-linux-x86_64-1.14.0.tar.gz -C "$BIN_DIR/tensorflow" >/dev/null 2>&1
fi
mv "$BIN_DIR/tensorflow/lib/libtensorflow.so.1" "$BIN_DIR/libs/libtensorflow.so.1" mv "$BIN_DIR/tensorflow/lib/"* "$BIN_DIR/libs/"
mv "$BIN_DIR/tensorflow/lib/libtensorflow_framework.so.2.6.0" "$BIN_DIR/libs/libtensorflow_framework.so.2" fi
# Build the wheel # Build the wheel
echo "Building the wheel..." echo "Building the wheel..."
@ -155,8 +166,6 @@ mv "$BIN_DIR/dist/"*.whl "$BUILD_DIR"
mv "$BIN_DIR/libs" "$BUILD_DIR" mv "$BIN_DIR/libs" "$BUILD_DIR"
mv "$BIN_DIR/plugins.xml" "$BUILD_DIR/libs" mv "$BIN_DIR/plugins.xml" "$BUILD_DIR/libs"
strip "$BUILD_DIR/libs"/*.so*
# Move the assets to the root directory # Move the assets to the root directory
mv "$SDK_DIR/assets" "$BUILD_DIR/assets" mv "$SDK_DIR/assets" "$BUILD_DIR/assets"