From 1416cd72323d0feaa59b6233a7b3ab9d22114e4b Mon Sep 17 00:00:00 2001 From: Nattapon Jaroenchai <nj7@illinois.edu> Date: Tue, 17 Oct 2023 14:29:56 -0500 Subject: [PATCH] Update README.md --- README.md | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/README.md b/README.md index 04b0447..dc2f8b7 100644 --- a/README.md +++ b/README.md @@ -25,17 +25,17 @@ The US's manufacturing and national security sectors heavily rely on a diverse r | File Name | Description | |-----------|-------------| -| `GAN_model.py` | V-Unet and GAN model script | +| `GAN_model.py` | Training GAN model for POC | | `README.md` | Documentation of the project | -| `VAE-unet.py` | VAE-Unet model script | +| `VAE-unet.py` | Create and train VAE U-net model for map segmentation | +| `VAE-unet_updated.py` | Create and train VAE U-net model for map segmentation with updated loss function | | `create_prediction_map.py` | Data loader class script for input processing | -| `data_util.py` | Utility functions for V-Unet and GAN model | +| `data_util.py` | Utility functions for dataloader that read and process data from TIFF and JSON files | | `eval_gan.py` | Evaluation script for GAN model | | `eval_vauner.py` | Evaluation script for VAE-Unet model | -| `train_model.py` | Training script for the model | -| `train_model_u_legends.py` | Training script with legends | -| `two-h-model.py` | Script for V-Unet and GAN model | -| `unet_util.py` | Utility functions for the Unet model | +| `train_model.py` | Training script for the attention model with processed legends| +| `train_model_u_legends.py` | Training script for the attention model with unprocess legends| +| `unet_util.py` | Utility functions for the attention U-net model | For more information on specific files or collaboration, please refer to the contribution guidelines or reach out to the repository maintainers. -- GitLab