diff --git a/README.md b/README.md index 6be808e63e3a158a8648f0fe84c6b1ede0bc6319..04b044788d1a044cf806994720cd0a8174bbb66d 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,50 @@ -# DARPA_proj +# DARPA CriticalMAAS Project -This project is created by Nattapon Jaroenchai (nj7@illinois.edu) +## Artificial Intelligence for Critical Mineral Assessment Competition + +The Defense Advanced Research Projects Agency (DARPA) and the United States Geological Survey (USGS) have joined hands in a groundbreaking endeavor to harness the potential of Machine Learning (ML) and Artificial Intelligence (AI) for bolstering the efficiency of critical mineral assessments. + +**Note:** The competition has concluded. DARPA's collaboration with USGS is still active under the Critical Mineral Assessments with AI Support (CriticalMAAS) project. + +For more details, visit [sam.gov](https://sam.gov). + + +*Salt flat at Death Valley | Source: [U.W. Geological Survey](https://www.usgs.gov)* + +## About Our Effort + + +*Mapping advanced argillic alteration at Cuprite, Nevada using imaging spectroscopy. +[Read the full publication](https://pubs.er.usgs.gov/publication/70196084)* + +The US's manufacturing and national security sectors heavily rely on a diverse range of non-fuel raw materials termed as "critical minerals." Both the Energy Act of 2020 and the Bipartisan Infrastructure Law have directed the USGS to carry out comprehensive assessments of these vital resources. These studies aim to pinpoint potential mineral resources from existing mines – both historical and active, offering insights into opportunities for sustainable resource development. Given the urgency of addressing modern-day supply chain needs, traditional methods are time-inefficient. This collaboration between DARPA and USGS is an attempt to supercharge these assessments using ML and AI. + +**Repository Status:** Ongoing development. Current files primarily consist of scripts for creating and training models focused on map segmentation. + +## Repo Contents + +| File Name | Description | +|-----------|-------------| +| `GAN_model.py` | V-Unet and GAN model script | +| `README.md` | Documentation of the project | +| `VAE-unet.py` | VAE-Unet model script | +| `create_prediction_map.py` | Data loader class script for input processing | +| `data_util.py` | Utility functions for V-Unet and GAN model | +| `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 | + +For more information on specific files or collaboration, please refer to the contribution guidelines or reach out to the repository maintainers. + + +## To-Do + +- [ ] **Evaluate VAE Unet model** +- [ ] **Incorporate intermediate input data** +- [ ] **Test color jittering for legends** +- [ ] **Legend augmentation** + +**Reference:** [Link to Paper](https://arxiv.org/abs/1804.04694)