config.json

{
    "model_name": "astronomy-image-classification-ensemble",
    "description": "Multi-class astronomy image classification system using ensemble of ResNet50 and DenseNet121",
    "classes": ["constellation", "cosmos", "galaxies", "nebula", "planets", "stars"],
    "input_size": [224, 224],
    "model_architecture": "ResNet50 + DenseNet121 Ensemble",
    "individual_accuracies": {
      "resnet50": 0.6486,
      "densenet121": 0.6396
    },
    "ensemble_expected_accuracy": "70-75%",
    "target_accuracy": 0.95,
    "framework": "PyTorch",
    "ensemble_method": "Soft Voting (Average Probabilities)",
    "preprocessing": {
      "resize": [224, 224],
      "normalization": "ImageNet",
      "augmentation": "Albumentations"
    },
    "model_files": [
      "best_resnet50.pth",
      "best_densenet121.pth"
    ]
  }