Project import generated by Copybara.

GitOrigin-RevId: ac03a471f5b9df34de46dd684202e4365c5ceac3

mediapipe/examples/desktop/youtube8m/README.md

@@ -33,7 +33,7 @@
 
 4.  Generate a MediaSequence metadata from the input video.
 
-    Note: the output file is /tmp/mediapipe/metadata.tfrecord
+    Note: the output file is /tmp/mediapipe/metadata.pb
 
     ```bash
     # change clip_end_time_sec to match the length of your video.
@@ -51,8 +51,17 @@
 
     GLOG_logtostderr=1 bazel-bin/mediapipe/examples/desktop/youtube8m/extract_yt8m_features \
       --calculator_graph_config_file=mediapipe/graphs/youtube8m/feature_extraction.pbtxt \
-      --input_side_packets=input_sequence_example=/tmp/mediapipe/metadata.tfrecord  \
-      --output_side_packets=output_sequence_example=/tmp/mediapipe/output.tfrecord
+      --input_side_packets=input_sequence_example=/tmp/mediapipe/metadata.pb  \
+      --output_side_packets=output_sequence_example=/tmp/mediapipe/features.pb
+    ```
+
+6.  [Optional] Read the features.pb in Python.
+
+    ```
+    import tensorflow as tf
+
+    sequence_example = open('/tmp/mediapipe/features.pb', 'rb').read()
+    print(tf.train.SequenceExample.FromString(sequence_example))
     ```
 
 ### Steps to run the YouTube-8M inference graph with the YT8M dataset
@@ -118,7 +127,7 @@
 
 ### Steps to run the YouTube-8M model inference graph with a local video
 
-1.  Make sure you have the output tfrecord from the feature extraction pipeline.
+1.  Make sure you have the features.pb from the feature extraction pipeline.
 
 2.  Copy the baseline model [(model card)](https://drive.google.com/file/d/1xTCi9-Nm9dt2KIk8WR0dDFrIssWawyXy/view) to local.
 
@@ -138,7 +147,7 @@
     # overlap is the number of seconds adjacent segments share.
     GLOG_logtostderr=1 bazel-bin/mediapipe/examples/desktop/youtube8m/model_inference \
       --calculator_graph_config_file=mediapipe/graphs/youtube8m/local_video_model_inference.pbtxt \
-      --input_side_packets=input_sequence_example_path=/tmp/mediapipe/output.tfrecord,input_video_path=/absolute/path/to/the/local/video/file,output_video_path=/tmp/mediapipe/annotated_video.mp4,segment_size=5,overlap=4
+      --input_side_packets=input_sequence_example_path=/tmp/mediapipe/features.pb,input_video_path=/absolute/path/to/the/local/video/file,output_video_path=/tmp/mediapipe/annotated_video.mp4,segment_size=5,overlap=4
     ```
 
 4.  View the annotated video.

mediapipe/examples/desktop/youtube8m/generate_input_sequence_example.py

@@ -53,7 +53,7 @@ def main(argv):
       flags.FLAGS.clip_start_time_sec * SECONDS_TO_MICROSECONDS, metadata)
   ms.set_clip_end_timestamp(
       flags.FLAGS.clip_end_time_sec * SECONDS_TO_MICROSECONDS, metadata)
-  with open('/tmp/mediapipe/metadata.tfrecord', 'wb') as writer:
+  with open('/tmp/mediapipe/metadata.pb', 'wb') as writer:
     writer.write(metadata.SerializeToString())