This commit is contained in:
2026-01-16 16:03:48 +08:00
parent 3c8299a188
commit 486772a85d
3 changed files with 85 additions and 1 deletions

View File

@ -4,7 +4,8 @@ import os
from rknnlite.api import RKNNLite
# ====================== 配置区 ======================
#calculate_diff2.py对应新版本calculate_diff3.py
#calculate_diff2.py对应新版本calculate_diff3.py
MODEL_PATH = "./Mv3D/point.rknn"
OUTPUT_DIR = "./output_rknn"
os.makedirs(OUTPUT_DIR, exist_ok=True)

Binary file not shown.

83
sh/ailai_app.sh Normal file
View File

@ -0,0 +1,83 @@
#!/bin/bash
LOG_FILE="/mnt/data/ailai-app/log/app_$(date +%Y%m%d_%H%M%S).log"
log_message() {
echo "$(date '+%Y-%m-%d %H:%M:%S') - $1" >> "$LOG_FILE"
}
wait_for_gui() {
log_message "wait gui..."
while [ ! -f "/home/teamhd/.Xauthority" ]; do
log_message "wait x11..."
sleep 2
done
export DISPLAY=:0
export XAUTHORITY="/home/teamhd/.Xauthority"
while [ ! xset q > /dev/null 2>&1]; do
log_message "cant connect x11"
sleep 1
done
log_message "x11 is ok "
return 0
}
setup_conda() {
log_message "set conda"
CONDA_PATH="/mnt/data/ailai/yes/etc/profile.d/conda.sh"
if [ -f "$CONDA_PATH" ]; then
source "$CONDA_PATH"
log_message "Conda load success"
else
log_message "conda fail"
return 1
fi
# sudo /mnt/data/ailai/yes/condabin/conda activate ailai
if [ $? -eq 0 ]; then
log_message "Conda activate success"
else
log_message "conda activate fail"
return 1
fi
return 0
}
main() {
log_message "app start"
if ! wait_for_gui; then
log_message "start gui fail"
exit 1
fi
# if ! setup_conda; then
# log_message "start conda fail"
# exit 1
# fi
cd "/mnt/data/ailai-app" || {
log_message "app dir fail"
exit 1
}
log_message "start python..."
sudo /mnt/data/ailai/yes/envs/ailai/bin/python main.py >> "$LOG_FILE" 2>&1
EXIT_CODE=$?
log_message "exit code:$EXIT_CODE"
log_message "app end"
}
main "$@"