Skip to content
Snippets Groups Projects
Commit 4f2dc0c8 authored by b.gong's avatar b.gong
Browse files

update video_prediction/datasets/era5_dataset_v2.py

parent 29cb2940
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
video_prediction/datasets/era5_dataset_v2.py
+ 1
− 1
View file @ 4f2dc0c8
...@@ -85,7 +85,7 @@ class ERA5Dataset_v2(VarLenFeatureVideoDataset): ...@@ -85,7 +85,7 @@ class ERA5Dataset_v2(VarLenFeatureVideoDataset):
# images = tf.decode_raw(parsed_features["images/encoded"],tf.int32) # images = tf.decode_raw(parsed_features["images/encoded"],tf.int32)
# images = seq # images = seq
images = tf.reshape(seq, [20, 64, 64, 3], name = "reshape_new") images = tf.reshape(seq, [20, 128, 160, 3], name = "reshape_new")
print("IMAGES", images) print("IMAGES", images)
seqs["images"] = images seqs["images"] = images
return seqs return seqs
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment