Galaxy-Images-Clustering

CV project - clustering low-res images, part of IAD at WUT

Installation step

1.    pip install -r requirements.txt

2. Run eda notebook to see the data distribution and some examples of images.3
3. Data preprocessing and features extraction using EfficientNet:

     python src/preprocessing.py --name train_mapping.csv
     python src/preprocessing.py --name validation_mapping.csv
     python src/preprocessing.py --name test_mapping.csv
     python src/features_efficientnet.py --name train_mapping.csv
     python src/features_efficientnet.py --name validation_mapping.csv
     python src/features_efficientnet.py --name test_mapping.csv

4. Run train notebook to train feature extractor architecture.

Sources

• paper
• paper
• paper
• paper
• video