Researching the saved models.

Author: rigolepe

.gitignore

@@ -4,6 +4,7 @@ output/*
 .idea
 _models/*
 _posenet_weights/*
+env.sh
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

Dockerfile

@@ -1,4 +1,5 @@
-FROM tensorflow/tensorflow:2.0.0-gpu-py3-jupyter
+FROM tensorflow/tensorflow:nightly-py3-jupyter
+# This is the CPU version!
 # see: https://www.tensorflow.org/install/docker
 # see: https://hub.docker.com/r/tensorflow/tensorflow/
 
@@ -7,6 +8,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
       bzip2 \
       git \
       wget && \
+    pip install --upgrade pip && \
     rm -rf /var/lib/apt/lists/*
 
 COPY requirements.txt /work/
@@ -15,6 +17,12 @@ WORKDIR /work
 
 RUN pip install -r requirements.txt
 
+RUN git clone https://github.com/patlevin/tfjs-to-tf.git && \
+    cd tfjs-to-tf && \
+    pip install . && \
+    cd .. && \
+    rm -r tfjs-to-tf
+
 ENV PYTHONPATH='/work/:$PYTHONPATH'
 
 CMD ["bash"]

Dockerfile-gpu

@@ -0,0 +1,27 @@
+FROM tensorflow/tensorflow:2.0.0-gpu-py3-jupyter
+# see: https://www.tensorflow.org/install/docker
+# see: https://hub.docker.com/r/tensorflow/tensorflow/
+
+# Install system packages
+RUN apt-get update && apt-get install -y --no-install-recommends \
+      bzip2 \
+      git \
+      wget && \
+    pip install --upgrade pip && \
+    rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt /work/
+
+WORKDIR /work
+
+RUN pip install -r requirements.txt
+
+RUN git clone https://github.com/patlevin/tfjs-to-tf.git && \
+    cd tfjs-to-tf && \
+    pip install . && \
+    cd .. && \
+    rm -r tfjs-to-tf
+
+ENV PYTHONPATH='/work/:$PYTHONPATH'
+
+CMD ["bash"]

docker_img_build.sh

@@ -1,5 +1,19 @@
 #!/usr/bin/env bash
 
-docker rmi -f posenet-python
+if [ -z "$1" ]; then
+  echo "pass CPU or GPU as argument"
+  echo "Docker image build failed..."
+  exit 1
+fi
 
-docker build -t posenet-python -f Dockerfile .
+if [ "$1" = "GPU" ]; then
+  image="posenet-python-gpu"
+  dockerfile="Dockerfile-gpu"
+else
+  image="posenet-python-cpu"
+  dockerfile="Dockerfile"
+fi
+
+docker rmi -f "$image"
+
+docker build -t "$image" -f "$dockerfile" .

docker_run.sh

@@ -2,4 +2,26 @@
 
 WORK=$(pwd)
 
-docker run --gpus all -it -v $WORK:/work posenet-python python "$@"
+if [ -z "$POSENET_PYTHON_DEVICE" ]; then
+  echo "set the environment variable POSENET_PYTHON_DEVICE to CPU or GPU, or enter your choice below:"
+  read -p "Enter your device (CPU or GPU): "  device
+  if [ "$device" = "GPU" ]; then
+    source <(echo "export POSENET_PYTHON_DEVICE=GPU");
+  elif [ "$device" = "CPU" ]; then
+    source <(echo "export POSENET_PYTHON_DEVICE=CPU");
+  else
+    echo "Device configuration failed..."
+    exit 1
+  fi
+fi
+
+
+echo "device is: $POSENET_PYTHON_DEVICE"
+
+if [ "$POSENET_PYTHON_DEVICE" = "GPU" ]; then
+  image="posenet-python-gpu"
+else
+  image="posenet-python-cpu"
+fi
+
+docker run --gpus all -it --rm -v $WORK:/work "$image" python "$@"

posenet/converter/tfjs2tf.py

@@ -0,0 +1,8 @@
+import tfjs_graph_converter as tfjs
+import os
+
+
+def convert(model_id, model_dir, check=False):
+    tfjsdir = os.path.join('/opt/project/_posenet_weights', 'mobilenet_v1_101')
+    tfdir = os.path.join('/opt/project/_models', 'model-mobilenet_v1_101_test')
+    tfjs.api.graph_model_to_saved_model(tfjsdir, tfdir, ['serve'])

requirements.txt

@@ -4,3 +4,5 @@ scipy==1.1.*
 pyyaml==3.*
 opencv-python-headless==3.4.5.20
 # opencv-python==3.4.5.20
+# tensorflowjs==1.3.2
+tensorflowjs==1.4.0

test_tfjs2tf.py

@@ -0,0 +1,28 @@
+import posenet.converter.tfjs2tf as converter
+
+
+def main():
+    converter.convert()
+
+# have a look at: https://github.com/tensorflow/tfjs/tree/master/tfjs-converter
+
+# see: https://stackoverflow.com/questions/58841355/bodypix-real-time-person-segmentation/59509874#59509874
+# https://github.com/patlevin/tfjs-to-tf
+# https://storage.googleapis.com/tfjs-models/savedmodel/bodypix/resnet50/float/model-stride16.json
+# see weightsManifest.paths for the shard names:
+# https://storage.googleapis.com/tfjs-models/savedmodel/bodypix/resnet50/float/group1-shard1of23.bin
+# ...
+# https://storage.googleapis.com/tfjs-models/savedmodel/bodypix/resnet50/float/group1-shard23of23.bin
+
+
+# https://storage.googleapis.com/tfjs-models/savedmodel/bodypix/mobilenet/float/100/model-stride16.json
+# see weightsManifest.paths
+# "group1-shard1of4.bin",
+# "group1-shard2of4.bin",
+# "group1-shard3of4.bin",
+# "group1-shard4of4.bin"
+
+
+# https://storage.googleapis.com/tfjs-models/weights/posenet/mobilenet_v1_101/manifest.json
+
+