mirror of
https://github.com/photoprism/photoprism.git
synced 2025-12-12 00:34:13 +01:00
Config: Change default vision model assets path to assets/models/ #127
Signed-off-by: Michael Mayer <michael@photoprism.app>
This commit is contained in:
@@ -4,7 +4,8 @@ TODAY=$(date -u +%Y%m%d)
|
||||
|
||||
MODEL_NAME="Facenet"
|
||||
MODEL_URL="https://dl.photoprism.app/tensorflow/facenet.zip?$TODAY"
|
||||
MODEL_PATH="assets/facenet"
|
||||
MODELS_PATH="assets/models"
|
||||
MODEL_PATH="$MODELS_PATH/facenet"
|
||||
MODEL_ZIP="/tmp/photoprism/facenet.zip"
|
||||
MODEL_HASH="0492eb1d67789108b7eefb274e26633504b059be $MODEL_ZIP"
|
||||
MODEL_VERSION="$MODEL_PATH/version.txt"
|
||||
@@ -17,7 +18,7 @@ mkdir -p /tmp/photoprism
|
||||
mkdir -p storage/backup
|
||||
|
||||
# Check for update
|
||||
if [[ -f ${MODEL_ZIP} ]] && [[ $(sha1sum ${MODEL_ZIP}) == ${MODEL_HASH} ]]; then
|
||||
if [[ -f ${MODEL_ZIP} ]] && [[ $(sha1sum ${MODEL_ZIP}) == "${MODEL_HASH}" ]]; then
|
||||
if [[ -f ${MODEL_VERSION} ]]; then
|
||||
echo "Already up to date."
|
||||
exit
|
||||
@@ -40,7 +41,7 @@ if [[ -e ${MODEL_PATH} ]]; then
|
||||
fi
|
||||
|
||||
# Unzip model
|
||||
unzip ${MODEL_ZIP} -d assets
|
||||
unzip ${MODEL_ZIP} -d "$MODELS_PATH"
|
||||
echo "$MODEL_NAME $TODAY $MODEL_HASH" > ${MODEL_VERSION}
|
||||
|
||||
echo "Latest $MODEL_NAME installed."
|
||||
|
||||
@@ -4,7 +4,8 @@ TODAY=$(date -u +%Y%m%d)
|
||||
|
||||
MODEL_NAME="NASNet Mobile"
|
||||
MODEL_URL="https://dl.photoprism.app/tensorflow/nasnet.zip?$TODAY"
|
||||
MODEL_PATH="assets/nasnet"
|
||||
MODELS_PATH="assets/models"
|
||||
MODEL_PATH="$MODELS_PATH/nasnet"
|
||||
MODEL_ZIP="/tmp/photoprism/nasnet.zip"
|
||||
MODEL_HASH="f18b801354e95cade497b4f12e8d2537d04c04f6 $MODEL_ZIP"
|
||||
MODEL_VERSION="$MODEL_PATH/version.txt"
|
||||
@@ -41,7 +42,7 @@ if [[ -e ${MODEL_PATH} ]]; then
|
||||
fi
|
||||
|
||||
# Unzip model
|
||||
unzip ${MODEL_ZIP} -d assets
|
||||
unzip ${MODEL_ZIP} -d "$MODELS_PATH"
|
||||
echo "$MODEL_NAME $TODAY $MODEL_HASH" > ${MODEL_VERSION}
|
||||
wget --inet4-only -c "${MODEL_21K_LABELS_URL}" -O ${MODEL_PATH}/labels21k.txt
|
||||
|
||||
|
||||
@@ -4,7 +4,8 @@ TODAY=$(date -u +%Y%m%d)
|
||||
|
||||
MODEL_NAME="NSFW"
|
||||
MODEL_URL="https://dl.photoprism.app/tensorflow/nsfw.zip?$TODAY"
|
||||
MODEL_PATH="assets/nsfw"
|
||||
MODELS_PATH="assets/models"
|
||||
MODEL_PATH="$MODELS_PATH/nsfw"
|
||||
MODEL_ZIP="/tmp/photoprism/nsfw.zip"
|
||||
MODEL_HASH="2e03ad3c6aec27c270c650d0574ff2a6291d992b $MODEL_ZIP"
|
||||
MODEL_VERSION="$MODEL_PATH/version.txt"
|
||||
@@ -40,7 +41,7 @@ if [[ -e ${MODEL_PATH} ]]; then
|
||||
fi
|
||||
|
||||
# Unzip model
|
||||
unzip ${MODEL_ZIP} -d assets
|
||||
unzip ${MODEL_ZIP} -d "$MODELS_PATH"
|
||||
echo "$MODEL_NAME $TODAY $MODEL_HASH" > ${MODEL_VERSION}
|
||||
|
||||
echo "Latest $MODEL_NAME installed."
|
||||
|
||||
Reference in New Issue
Block a user