Skip to content

GitLab

"...resnet50_tensorflow.git" did not exist on "6ef140dcd0348bd6cea5d59a9ed807af0f3e9040"
Commit b2d1ef12 authored by Liangzhe Yuan's avatar Liangzhe Yuan Committed by A. Unique TensorFlower
Browse files

Fix the audio dimensions for eval input. 

PiperOrigin-RevId: 420825486
parent 69dd34aa
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/dataloaders/video_input.py
View file @ b2d1ef12
...@@ -361,7 +361,7 @@ class Parser(parser.Parser): ...@@ -361,7 +361,7 @@ class Parser(parser.Parser):
audio = decoded_tensors[self._audio_feature] audio = decoded_tensors[self._audio_feature]
audio = tf.cast(audio, dtype=self._dtype) audio = tf.cast(audio, dtype=self._dtype)
audio = preprocess_ops_3d.sample_sequence( audio = preprocess_ops_3d.sample_sequence(
audio, 20, random=False, stride=1) audio, self._audio_shape[0], random=False, stride=1)
audio = tf.ensure_shape(audio, self._audio_shape) audio = tf.ensure_shape(audio, self._audio_shape)
features['audio'] = audio features['audio'] = audio
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment