juppy44/galaxy-zoo-resnet18

Galaxy Zoo ResNet18

Fine-tuned ResNet18 for Galaxy Zoo morphology classification.

Classes

• Smooth Round
• Smooth Cigar
• Edge-on Disk
• Spiral (bar)
• Spiral (no bar)
• Irregular
• Merger
• Artifact/Star

Files

• galaxy_zoo_resnet18.pth: trained PyTorch checkpoint
• class_names.json: label names
• inference.py: helper code for loading and inference

Notes

This repo stores a plain PyTorch checkpoint, not a Transformers save_pretrained() model.