{ "model_version": "1.0.0", "data": { "species": "Mus musculus", "strain": "C57BL/6J", "sexes": ["female", "male"], "age_group": "adult", "imaging": "epifluorescence microscopy", "sample_channel": "green channel, EGFP spectral range", "atlas_id": "allen_mouse_10um", "atlas_version": "1.2", "atlas_resolution_um": 10.0, "image_width_px": 1140, "image_height_px": 800, "total_animals": 87, "total_sections": 5613, "split_unit": "animal", "split_seed": 42, "splits": { "train": {"animals": 61, "sections": 4015}, "validation": {"animals": 13, "sections": 745}, "test": {"animals": 13, "sections": 853} }, "target_registration_software": "SHARPy-track within DMC-BrainMap", "target_reviewers": ["Felix Jung", "Xiao Cao", "Loran Heymans"], "acceptable_mean_tre_px": 4.0, "acceptable_mean_tre_um": 40.0, "maximum_landmark_homography_fit_rmse_px": 60.0, "excluded_above_fit_rmse": 296, "training_images_public": false }, "model": { "model_name": "spatial", "base_channels": 24, "spatial_pool_size": [5, 8], "spatial_hidden_channels": 256, "max_corner_offset_px": 512.0, "parameter_count": 2336200 }, "optimization": { "maximum_epochs": 200, "stopped_after_epoch": 111, "selected_epoch": 91, "selection_metric": "validation_loss", "selected_validation_loss": 16.35543269727054, "batch_size": 4, "optimizer": "AdamW", "learning_rate": 0.0001, "weight_decay": 0.0001, "scheduler": "ReduceLROnPlateau", "scheduler_factor": 0.5, "scheduler_patience": 6, "minimum_learning_rate": 5e-7, "early_stopping_patience": 20, "early_stopping_min_delta": 0.03, "seed": 42 }, "loss": { "primary": "Smooth L1 reprojection loss", "points_per_axis": 8, "sample_grid_points": 64, "sample_bbox_inset_fraction": 0.02, "corner_loss": "Smooth L1", "corner_loss_weight": 0.25 }, "augmentation": { "applied_splits": ["train"], "preset": "moderate", "seed": 42, "changes_geometry": false, "missing_chunk_probability": 0.45, "missing_chunk_count": [1, 3], "missing_chunk_area_fraction": [0.02, 0.10], "tear_probability": 0.35, "crack_probability": 0.30, "fold_probability": 0.30, "exposure_probability": 0.65, "brightness_delta": [-22.0, 22.0], "contrast_factor": [0.75, 1.35], "gamma": [0.70, 1.45], "noise_sigma": [0.0, 8.0] }, "environment": { "python": "3.14", "pytorch": "2.12.0+cu130", "torchvision": "0.27.0+cu130", "numpy": "2.4.4", "opencv_python": "4.13.0.92", "gpu": "NVIDIA GeForce RTX 3070 Ti" }, "provenance": { "source_repository": "https://github.com/xiao-1011/sharpy-model", "source_commit": "0a80f4f831f6450a015a2f32433f22975ac97dc0", "run_name": "baseline-005-spatial-augmented-200ep", "original_checkpoint_sha256": "2f6c84b12a8449a63ff45973cb91a81326973915d4801659c99b973087c1dac8" } }