MCPcopy Create free account

hub / github.com/Chihiro-n/csiro-biomass-agentic-solution / types & classes

Types & classes64 in github.com/Chihiro-n/csiro-biomass-agentic-solution

↓ 3 callersClassBiomassDatasetOOF
Dataset for OOF prediction - processes both halves separately but tracks original image. Returns: (image_tensor_left, image_tensor_r
EXP/EXP060/train.py:1625
↓ 3 callersClassEVACLIPBiomassModel5Head
EVA-CLIP-based model with 5 separate heads for all targets. Architecture: 1. EVA-CLIP Vision Encoder (768-dim embeddings for EVA02-C
EXP/EXP060/train.py:2033
↓ 3 callersClassTimmBiomassModel5Head
Timm-based model with 5 separate heads for all targets. Architecture: 1. timm Vision Model (e.g., eva02_large_patch14_448.mim_m38m_f
EXP/EXP060/train.py:2324
↓ 3 callersClassTimmBiomassModel5HeadHeightFiLM
Timm-based 5-head model with Height prediction and FiLM conditioning. Key innovation (kmat-style): - Predicts Height from image (auxilia
EXP/EXP060/train.py:2457
↓ 2 callersClassBiomassDatasetOOF
Dataset for OOF prediction - processes both halves separately but tracks original image. Returns: (image_tensor_left, image_tensor_r
EXP/EXP113/train.py:393
↓ 2 callersClassBiomassDatasetOOF3Crop
Dataset for 3-crop OOF prediction. Returns all 3 crops for each image: - Left: 0-1000 - Center: 500-1500 - Right: 1000-2000
EXP/EXP060/train.py:1431
↓ 2 callersClassBiomassDatasetSplit
Training Dataset with image splitting and labels. Each 2000x1000 image is split into left and right halves. Labels are HALVED since each
EXP/EXP060/infer_ttt.py:105
↓ 2 callersClassEVACLIPBiomassModel
EVA-CLIP-based model with gradual unfreezing capability. Architecture: 1. EVA-CLIP Vision Encoder (768-dim embeddings for EVA02-CLIP
EXP/EXP060/train.py:1883
↓ 2 callersClassEVACLIPBiomassModel5HeadBinning
EVA-CLIP model with 5 binning heads for biomass prediction. Each head outputs logits for bin classification. At inference, converts to e
EXP/EXP060/train.py:3314
↓ 2 callersClassInternViTBiomassModel5Head
InternViT-based model with 5 separate heads for all targets. Uses LOCAL InternViT implementation (no flash_attn dependency). Weights are
EXP/EXP060/train.py:3048
↓ 2 callersClassOpenCLIPBiomassModel5Head
OpenCLIP-based model with 5 separate heads for all targets. Uses open_clip to load models from HuggingFace Hub. This is required for mod
EXP/EXP060/train.py:2925
↓ 2 callersClassTimmBiomassModel
Timm-based model with gradual unfreezing capability (3-head version). Architecture: 1. timm Vision Model (e.g., eva02_large_patch14_
EXP/EXP060/train.py:2245
↓ 2 callersClassTimmBiomassModel5HeadDensity
Timm-based model with Residual Density Head. Architecture: 1. DINOv3 ViT backbone: - CLS token → Global prediction (stabl
EXP/EXP113/train.py:546
↓ 2 callersClassTimmBiomassModel5HeadGaussian
Timm-based model with Gaussian output for uncertainty-aware predictions. Instead of predicting point estimates, this model outputs mean (mu)
EXP/EXP060/train.py:2657
↓ 2 callersClassTimmBiomassModel5HeadSoftSpecies
Timm-based model with Soft-Species Conditioning. Architecture: 1. timm Vision Model → global pooled embedding (embed_dim) 2.
EXP/EXP060/train.py:2803
↓ 2 callersClassTimmBiomassModelDensityOnly
Pure Density Integration Model (No Global Head). Architecture: 1. DINOv3 ViT backbone extracts patch tokens (32×32 for 512px input)
EXP/EXP113/train.py:739
↓ 2 callersClassWeightedBiomassLoss5Head
Weighted SmoothL1Loss for 5-head model with consistency loss.
EXP/EXP060/infer_ttt.py:275
↓ 1 callersClassBinningLoss
CrossEntropyLoss for binning classification. Converts continuous targets to bin indices and computes weighted CE loss. Args: r2
EXP/EXP060/train.py:3482
↓ 1 callersClassBiomassDatasetMosaic
Dataset with Mosaic augmentation for biomass prediction. Mosaic combines 4 images into a 2x2 grid: - Each image is resized to half size
EXP/EXP060/train.py:1665
↓ 1 callersClassBiomassDatasetOOFTTA
Dataset for OOF prediction with TTA support. Returns 4 views for each image: - Original - HorizontalFlip - VerticalFlip - HF
EXP/EXP060/train.py:1560
↓ 1 callersClassBiomassDatasetOOFTTA
Dataset for OOF prediction with TTA support. Returns 4 views for each image: - Original - HorizontalFlip - VerticalFlip - HF
EXP/EXP113/train.py:486
↓ 1 callersClassBiomassDatasetSplit
Dataset for EVA-CLIP-based biomass prediction with image splitting. Each 2000x1000 image is split into: - Left half: 1000x1000 - Rig
EXP/EXP060/train.py:1080
↓ 1 callersClassBiomassDatasetSplit
Dataset for biomass prediction with image splitting. Each 2000x1000 image is split into: - Left half: 1000x1000 - Right half: 1000x1
EXP/EXP113/train.py:225
↓ 1 callersClassBiomassDatasetSplit3Crop
Dataset for 3-crop training with overlapping crops. Each 2000x1000 image is split into 3 overlapping crops: - Left: 0-1000 (50% of i
EXP/EXP060/train.py:1338
↓ 1 callersClassBiomassDatasetSplitSumLoss
Dataset for sum-loss training (full-label supervision). Each sample returns BOTH halves together so training can compute: pred_full
EXP/EXP060/train.py:1244
↓ 1 callersClassBiomassDatasetSplitViewConsistency
Dataset for View Consistency training. For each sample, returns: - Original view (with random augmentation) - Flipped view (HFlip or
EXP/EXP060/train.py:1480
↓ 1 callersClassConsistencyLoss
Weighted loss with consistency constraints for biomass prediction. Enforces physical constraints: 1. Dry_Dead_g + GDM_g ≈ Dry_Total_
EXP/EXP060/train.py:3560
↓ 1 callersClassConsistencyLoss
Combined loss for 5-head biomass model with physical consistency constraints. Loss = SmoothL1Loss(predictions, targets) + consistency_weight
EXP/EXP113/train.py:1351
↓ 1 callersClassEVACLIPBiomassModel
EVA-CLIP-based Model with Frozen Embeddings and Trainable Linear Heads. Must match EXP/EXP030/train.py architecture exactly.
EXP/EXP060/infer.py:520
↓ 1 callersClassEVACLIPBiomassModel5Head
EVA-CLIP-based model with 5 separate heads for all targets (inference version). Architecture: 1. EVA-CLIP Vision Encoder (768-dim em
EXP/EXP060/infer.py:649
↓ 1 callersClassEVACLIPBiomassModel5Head
EVA-CLIP-based model with 5 separate heads for all targets.
EXP/EXP060/infer_ttt.py:185
↓ 1 callersClassGaussianConsistencyLoss
Gaussian NLL Loss with consistency constraints for biomass prediction. Instead of predicting point estimates, the model predicts mean (mu) a
EXP/EXP060/train.py:3700
↓ 1 callersClassGradientReversalLayer
Gradient Reversal Layer wrapper. Usage: grl = GradientReversalLayer(lambda_=0.1) reversed_features = grl(features)
EXP/EXP060/train.py:125
↓ 1 callersClassGradientReversalLayer
Gradient Reversal Layer wrapper. Usage: grl = GradientReversalLayer(lambda_=0.1) reversed_features = grl(features)
EXP/EXP113/train.py:99
↓ 1 callersClassRankingLoss
Ranking Loss for learning relative ordering (A > B, A < B). Key insight: SigLIP/CLIP models use semantic shortcuts ("Clover present" → predi
EXP/EXP060/train.py:3789
↓ 1 callersClassSpatialDeepDensityHeadFast
Spatial-Deep Density Head for fast inference.
EXP/EXP113/infer_fast.py:78
↓ 1 callersClassStratifiedBatchSampler
Batch sampler that ensures value diversity within each batch. For ranking loss to be effective, each batch should contain samples with d
EXP/EXP060/train.py:3902
↓ 1 callersClassTestBiomassDatasetSplit
Test Dataset for EVA-CLIP-based pipeline with image splitting. Each 2000x1000 image is split into: - Left half: (0, 0, 1000, 1000) -
EXP/EXP060/infer.py:89
↓ 1 callersClassTestBiomassDatasetSplit
Test Dataset for inference (no labels).
EXP/EXP060/infer_ttt.py:78
↓ 1 callersClassTestBiomassDatasetSplit
Test Dataset for timm-based pipeline with image splitting. Each 2000x1000 image is split into: - Left half: (0, 0, 1000, 1000) - Rig
EXP/EXP113/infer.py:45
↓ 1 callersClassTestBiomassDatasetSplit
Test Dataset with image splitting (No TTA).
EXP/EXP113/infer_fast.py:45
↓ 1 callersClassTestBiomassDatasetSplit3Crop
Test Dataset for 3-crop inference (matches train.py BiomassDatasetOOF3Crop). Each 2000x1000 image is split into 3 overlapping crops: - L
EXP/EXP060/infer.py:478
↓ 1 callersClassTestBiomassDatasetSplitBlurOnly
Test Dataset with blur preprocessing (no TTA, single view). Rationale: - Train images have varying sharpness (Laplacian variance: 181-10
EXP/EXP060/infer.py:345
↓ 1 callersClassTestBiomassDatasetSplitBlurTTA
Test Dataset with blur-based TTA (no flip/rotate). Rationale: - Train images have varying sharpness (Laplacian variance: 181-10030, 55x
EXP/EXP060/infer.py:404
↓ 1 callersClassTestBiomassDatasetSplitCropTTA
Test Dataset with crop-based TTA (physically consistent for this pipeline). Motivation: - Training uses RandomResizedCrop in some augmen
EXP/EXP060/infer.py:192
↓ 1 callersClassTestBiomassDatasetSplitPhotoTTA
Test Dataset with photometric TTA (no flip/rotate). Rationale: - In this competition, the hidden test split differs by Sampling_Date, wh
EXP/EXP060/infer.py:269
↓ 1 callersClassTestBiomassDatasetSplitTTA
Test Dataset with TTA (Test Time Augmentation) support. TTA applies 4 augmentations to each image: - Original - HorizontalFlip -
EXP/EXP060/infer.py:129
↓ 1 callersClassTestBiomassDatasetSplitTTA
Test Dataset with TTA (Test Time Augmentation) support. TTA applies 4 augmentations to each image: - Original - HorizontalFlip -
EXP/EXP113/infer.py:83
↓ 1 callersClassTimmBiomassModel
Timm-based Model with Frozen Embeddings and Trainable Linear Heads. Must match the EXP060 timm training architecture (3-head).
EXP/EXP060/infer.py:858
↓ 1 callersClassTimmBiomassModel5Head
Timm-based model with 5 separate heads for all targets (inference version).
EXP/EXP060/infer.py:915
↓ 1 callersClassTimmBiomassModel5HeadDensity
Timm-based model with Residual Density Head (inference version). Architecture: 1. DINOv3 ViT backbone: - CLS token → Glob
EXP/EXP113/infer.py:150
↓ 1 callersClassTimmBiomassModel5HeadGaussian
Timm-based model with Gaussian output for uncertainty-aware predictions (inference version). This model outputs mean (mu) for each target. V
EXP/EXP060/infer.py:1088
↓ 1 callersClassTimmBiomassModel5HeadHeightFiLM
Timm-based 5-head model with Height prediction and FiLM conditioning (inference version). Key innovation (kmat-style): - Predicts Height
EXP/EXP060/infer.py:1172
↓ 1 callersClassTimmBiomassModel5HeadSoftSpecies
Timm-based model with Soft-Species Conditioning (inference version). Architecture: 1. timm Vision Model → global pooled embedding (e
EXP/EXP060/infer.py:985
↓ 1 callersClassTimmBiomassModelDensityOnly
Pure Density Integration Model (No Global Head) - Inference version. Architecture: 1. DINOv3 ViT backbone extracts patch tokens (32×
EXP/EXP113/infer.py:298
↓ 1 callersClassTimmBiomassModelDensityOnlyFast
Pure Density Integration Model optimized for fast inference. - Supports FP16 inference - Minimal overhead
EXP/EXP113/infer_fast.py:102
↓ 1 callersClassWeightedBiomassLoss
Weighted SmoothL1Loss for biomass prediction.
EXP/EXP060/train.py:2010
ClassBiomassDatasetOOF
Validation Dataset - returns both halves for OOF prediction.
EXP/EXP060/infer_ttt.py:151
ClassGradientReversalFunction
Gradient Reversal Layer (GRL) for Domain Adversarial Training. Forward pass: identity Backward pass: negate gradients and scale by lambd
EXP/EXP060/train.py:104
ClassGradientReversalFunction
Gradient Reversal Layer (GRL) for Domain Adversarial Training. Forward pass: identity Backward pass: negate gradients and scale by lambd
EXP/EXP113/train.py:78
ClassResidualDilatedDensityHead
Residual density head: stable 1×1 baseline + small dilated 3×3 context correction. Starts identical to the 1×1 baseline by zero-init
EXP/EXP113/train.py:757
ClassResidualDilatedDensityHead
Residual density head: stable 1×1 baseline + small dilated 3×3 context correction.
EXP/EXP113/infer.py:312
ClassSpatialDeepDensityHead
Spatial-Deep Density Head: DWConv for spatial context + MLP for non-linearity. Architecture: LayerNorm → 3x3 DWConv → Ba
EXP/EXP113/train.py:803
ClassSpatialDeepDensityHead
Spatial-Deep Density Head: DWConv for spatial context + MLP for non-linearity. Inference version (no initialization needed - weights
EXP/EXP113/infer.py:346