Instructions to use MITCriticalData/Sentinel-2_Resnet50V2_VariationalAutoencoder_RGB_full_Colombia_Dataset with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Keras
How to use MITCriticalData/Sentinel-2_Resnet50V2_VariationalAutoencoder_RGB_full_Colombia_Dataset with Keras:
# Available backend options are: "jax", "torch", "tensorflow". import os os.environ["KERAS_BACKEND"] = "jax" import keras model = keras.saving.load_model("hf://MITCriticalData/Sentinel-2_Resnet50V2_VariationalAutoencoder_RGB_full_Colombia_Dataset") - Notebooks
- Google Colab
- Kaggle
Commit ·
36dafc0
1
Parent(s): 708409e
Update README.md
Browse files
README.md
CHANGED
|
@@ -4,15 +4,22 @@ library_name: keras
|
|
| 4 |
|
| 5 |
## Model description
|
| 6 |
|
| 7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
## Intended uses & limitations
|
| 10 |
|
| 11 |
-
|
| 12 |
|
| 13 |
## Training and evaluation data
|
| 14 |
|
| 15 |
-
|
|
|
|
|
|
|
| 16 |
|
| 17 |
## Training procedure
|
| 18 |
|
|
|
|
| 4 |
|
| 5 |
## Model description
|
| 6 |
|
| 7 |
+
Variational Autoencoder model trained to compress information from sentinel-2 satellite images using Resnet50 V2 as encoder backbone to extract features.
|
| 8 |
+
The latent space of the model is given by 1024 neurons which can be used to generate embeddings from the sentinel-2 satellite images.
|
| 9 |
+
|
| 10 |
+
The model was trained using bands RGB (2, 3 and 4) (Red, Green and Blue) of the Sentinel-2 satellites and using 81 municipalities of Colombia with most dengue cases.
|
| 11 |
+
|
| 12 |
+
The input shape of the model is 224, 224, 3. To extract features you should remove the last layer.
|
| 13 |
|
| 14 |
## Intended uses & limitations
|
| 15 |
|
| 16 |
+
The model was trained with images of 81 different cities in Colombia with most dengue cases, however it may require fine tuning or retraining to learn from other contexts such as countries and other continents.
|
| 17 |
|
| 18 |
## Training and evaluation data
|
| 19 |
|
| 20 |
+
The model was trained with satellite images of 81 different cities in Colombia extracted from sentinel-2 using RGB bands using an asymmetric variational autoencoder. Images with information that could result in noise such as black images were filtered prior to training to avoid noise in the data.
|
| 21 |
+
|
| 22 |
+
The dataset was split into train and test using 80% for train and 20% to test.
|
| 23 |
|
| 24 |
## Training procedure
|
| 25 |
|