API Reference
Package src/astrovision (import as src.astrovision from the repository root, as the notebooks do). Signatures below are read from the source; the code is the reference when they diverge.
Documented modules
| Page | Module | Main objects |
|---|---|---|
| models.py | models.py | SimpleCNN, EfficientNetGalaxy, build_model |
| trainer.py | trainer.py | train |
| morphometry.py | morphometry.py | GalaxyMorphometry, batch_morphometry, plot helpers |
| gradcam.py | gradcam.py | GradCAM, plot_gradcam_grid |
| crossmatcher.py | crossmatcher.py | Galaxy10Crossmatcher |
Modules without a dedicated page
| Module | Content | Where it is described |
|---|---|---|
data_loader.py | download_galaxy10, Galaxy10Dataset, get_transforms, load_galaxy10_splits, CLASS_NAMES | Data Preparation |
fusion.py | GalaxyFusionClassifier, FusionDataset, build_fusion_datasets, train_fusion, evaluate_fusion, ablation_study | Late-Fusion |
uncertainty.py | MCDropoutPredictor, TemperatureScaler, ConformalPredictor, find_most_uncertain | Uncertainty |
unet.py, segmenter.py | UNet, CombinedLoss, train_unet; OtsuSegmenter, DINOv2PatchSegmenter, SAMSegmenter | U-Net, Run Segmentation |
attention_morphometry.py | extract_attention_map, attention_to_mask, AttentionMorphometry | Attention |
synthesis.py, spectral_viz.py | master table, correlation and synthesis figures, colour–magnitude plots | Physical Interpretation |
src/astrovision/__init__.py re-exports load_galaxy10_splits, CLASS_NAMES, download_galaxy10, SimpleCNN, EfficientNetGalaxy, build_model, train, GradCAM and plot_gradcam_grid.
Related pages