google/balloon-learning-environment

How to use the ble_wind_field/small dataset in BLE for training a VAE?

Closed this issue · 0 comments

How to use the ble_wind_field/small dataset in BLE for training a VAE?
What is the purpose of this dataset?
How should it be used?

I am working with the Balloon Learning Environment (BLE) and I need to use the ble_wind_field/small dataset. I would like to know if this dataset can be used for training a Variational Autoencoder (VAE), and if so, how to write the code to achieve this.

  1. Installed BLE as per the official documentation.
  2. I ‘ve already Loaded the dataset using the following code

import numpy as np
import tensorflow as tf
import tensorflow_datasets as tfds

ds = tfds.load('ble_wind_field/small', split='train', shuffle_files=True, data_dir='datasets')
assert isinstance(ds, tf.data.Dataset)
print(ds)

ds = ds.take(1) # Only take a single example
len = 0
for i in tfds.as_numpy(ds):
print(i['field'].shape)
print(i)
len += 1
print('data length:',len)

Specific Questions

  1. Can the ble_wind_field/small dataset be used for training a VAE?
  2. If so, how can I write the code to preprocess this dataset and train a VAE?

Additional Information

  • Any code snippets or examples would be greatly appreciated.
  • If there are any specific files or configurations needed, please let me know.

Thank you for your help!