From 26813e25b3517b429ee20d87fc50c09c08e03965 Mon Sep 17 00:00:00 2001 From: Matthew Liu <96169489+TheMattBin@users.noreply.github.com> Date: Sun, 26 May 2024 20:52:42 +0800 Subject: [PATCH] Update input_pipeline.py --- scenic/projects/owl_vit/preprocessing/input_pipeline.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/scenic/projects/owl_vit/preprocessing/input_pipeline.py b/scenic/projects/owl_vit/preprocessing/input_pipeline.py index b9f5b962..12129e28 100644 --- a/scenic/projects/owl_vit/preprocessing/input_pipeline.py +++ b/scenic/projects/owl_vit/preprocessing/input_pipeline.py @@ -62,10 +62,11 @@ NUM_PARALLEL_CALLS = tf.data.AUTOTUNE +# Change decoders when version changes DECODERS = { 'visual_genome:1.0.0': label_ops.DecodeVisualGenome, - 'lvis:1.2.0': + 'lvis:1.3.0': label_ops.DecodeLvis, 'objects365:0.0.1': label_ops.DecodeObjects365,