From 66e5aa902b2f22d114722ed0eb520070227fd4ac Mon Sep 17 00:00:00 2001 From: Omar Sayed Mostafa Date: Sun, 9 Jan 2022 14:55:56 +0200 Subject: [PATCH] change head to commit 374f5e2, added yt8m feature extraction and inference build commands to dockerbuild file --- Dockerfile | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/Dockerfile b/Dockerfile index c4c4df3e4..e663b9015 100644 --- a/Dockerfile +++ b/Dockerfile @@ -66,6 +66,33 @@ azel-${BAZEL_VERSION}-installer-linux-x86_64.sh" && \ rm -f /bazel/installer.sh COPY . /mediapipe/ +WORKDIR /mediapipe/ +RUN mkdir /tmp/mediapipe +WORKDIR /tmp/mediapipe +RUN curl -O http://data.yt8m.org/pca_matrix_data/inception3_mean_matrix_data.pb +RUN curl -O http://data.yt8m.org/pca_matrix_data/inception3_projection_matrix_data.pb +RUN curl -O http://data.yt8m.org/pca_matrix_data/vggish_mean_matrix_data.pb +RUN curl -O http://data.yt8m.org/pca_matrix_data/vggish_projection_matrix_data.pb +RUN curl -O http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz +RUN tar -xvf /tmp/mediapipe/inception-2015-12-05.tgz + +WORKDIR /mediapipe/ +RUN python -m mediapipe.examples.desktop.youtube8m.generate_vggish_frozen_graph + +RUN bazel build -c opt --linkopt=-s \ + --define MEDIAPIPE_DISABLE_GPU=1 --define no_aws_support=true \ + mediapipe/examples/desktop/youtube8m:extract_yt8m_features + +RUN curl -o /tmp/mediapipe/yt8m_baseline_saved_model.tar.gz http://data.yt8m.org/models/baseline/saved_model.tar.gz + +RUN tar -xvf /tmp/mediapipe/yt8m_baseline_saved_model.tar.gz -C /tmp/mediapipe + +RUN bazel build -c opt --define='MEDIAPIPE_DISABLE_GPU=1' --linkopt=-s \ + mediapipe/examples/desktop/youtube8m:model_inference + + +RUN apt-get update +RUN apt-get install nano # If we want the docker image to contain the pre-built object_detection_offline_demo binary, do the following # RUN bazel build -c opt --define MEDIAPIPE_DISABLE_GPU=1 mediapipe/examples/desktop/demo:object_detection_tensorflow_demo