Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Chihiro-n/csiro-biomass-agentic-solution
/ types & classes
Types & classes
64 in github.com/Chihiro-n/csiro-biomass-agentic-solution
⨍
Functions
297
◇
Types & classes
64
↓ 3 callers
Class
BiomassDatasetOOF
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 callers
Class
EVACLIPBiomassModel5Head
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 callers
Class
TimmBiomassModel5Head
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 callers
Class
TimmBiomassModel5HeadHeightFiLM
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 callers
Class
BiomassDatasetOOF
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 callers
Class
BiomassDatasetOOF3Crop
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 callers
Class
BiomassDatasetSplit
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 callers
Class
EVACLIPBiomassModel
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 callers
Class
EVACLIPBiomassModel5HeadBinning
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 callers
Class
InternViTBiomassModel5Head
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 callers
Class
OpenCLIPBiomassModel5Head
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 callers
Class
TimmBiomassModel
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 callers
Class
TimmBiomassModel5HeadDensity
Timm-based model with Residual Density Head. Architecture: 1. DINOv3 ViT backbone: - CLS token → Global prediction (stabl
EXP/EXP113/train.py:546
↓ 2 callers
Class
TimmBiomassModel5HeadGaussian
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 callers
Class
TimmBiomassModel5HeadSoftSpecies
Timm-based model with Soft-Species Conditioning. Architecture: 1. timm Vision Model → global pooled embedding (embed_dim) 2.
EXP/EXP060/train.py:2803
↓ 2 callers
Class
TimmBiomassModelDensityOnly
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 callers
Class
WeightedBiomassLoss5Head
Weighted SmoothL1Loss for 5-head model with consistency loss.
EXP/EXP060/infer_ttt.py:275
↓ 1 callers
Class
BinningLoss
CrossEntropyLoss for binning classification. Converts continuous targets to bin indices and computes weighted CE loss. Args: r2
EXP/EXP060/train.py:3482
↓ 1 callers
Class
BiomassDatasetMosaic
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 callers
Class
BiomassDatasetOOFTTA
Dataset for OOF prediction with TTA support. Returns 4 views for each image: - Original - HorizontalFlip - VerticalFlip - HF
EXP/EXP060/train.py:1560
↓ 1 callers
Class
BiomassDatasetOOFTTA
Dataset for OOF prediction with TTA support. Returns 4 views for each image: - Original - HorizontalFlip - VerticalFlip - HF
EXP/EXP113/train.py:486
↓ 1 callers
Class
BiomassDatasetSplit
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 callers
Class
BiomassDatasetSplit
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 callers
Class
BiomassDatasetSplit3Crop
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 callers
Class
BiomassDatasetSplitSumLoss
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 callers
Class
BiomassDatasetSplitViewConsistency
Dataset for View Consistency training. For each sample, returns: - Original view (with random augmentation) - Flipped view (HFlip or
EXP/EXP060/train.py:1480
↓ 1 callers
Class
ConsistencyLoss
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 callers
Class
ConsistencyLoss
Combined loss for 5-head biomass model with physical consistency constraints. Loss = SmoothL1Loss(predictions, targets) + consistency_weight
EXP/EXP113/train.py:1351
↓ 1 callers
Class
EVACLIPBiomassModel
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 callers
Class
EVACLIPBiomassModel5Head
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 callers
Class
EVACLIPBiomassModel5Head
EVA-CLIP-based model with 5 separate heads for all targets.
EXP/EXP060/infer_ttt.py:185
↓ 1 callers
Class
GaussianConsistencyLoss
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 callers
Class
GradientReversalLayer
Gradient Reversal Layer wrapper. Usage: grl = GradientReversalLayer(lambda_=0.1) reversed_features = grl(features)
EXP/EXP060/train.py:125
↓ 1 callers
Class
GradientReversalLayer
Gradient Reversal Layer wrapper. Usage: grl = GradientReversalLayer(lambda_=0.1) reversed_features = grl(features)
EXP/EXP113/train.py:99
↓ 1 callers
Class
RankingLoss
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 callers
Class
SpatialDeepDensityHeadFast
Spatial-Deep Density Head for fast inference.
EXP/EXP113/infer_fast.py:78
↓ 1 callers
Class
StratifiedBatchSampler
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 callers
Class
TestBiomassDatasetSplit
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 callers
Class
TestBiomassDatasetSplit
Test Dataset for inference (no labels).
EXP/EXP060/infer_ttt.py:78
↓ 1 callers
Class
TestBiomassDatasetSplit
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 callers
Class
TestBiomassDatasetSplit
Test Dataset with image splitting (No TTA).
EXP/EXP113/infer_fast.py:45
↓ 1 callers
Class
TestBiomassDatasetSplit3Crop
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 callers
Class
TestBiomassDatasetSplitBlurOnly
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 callers
Class
TestBiomassDatasetSplitBlurTTA
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 callers
Class
TestBiomassDatasetSplitCropTTA
Test Dataset with crop-based TTA (physically consistent for this pipeline). Motivation: - Training uses RandomResizedCrop in some augmen
EXP/EXP060/infer.py:192
↓ 1 callers
Class
TestBiomassDatasetSplitPhotoTTA
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 callers
Class
TestBiomassDatasetSplitTTA
Test Dataset with TTA (Test Time Augmentation) support. TTA applies 4 augmentations to each image: - Original - HorizontalFlip -
EXP/EXP060/infer.py:129
↓ 1 callers
Class
TestBiomassDatasetSplitTTA
Test Dataset with TTA (Test Time Augmentation) support. TTA applies 4 augmentations to each image: - Original - HorizontalFlip -
EXP/EXP113/infer.py:83
↓ 1 callers
Class
TimmBiomassModel
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 callers
Class
TimmBiomassModel5Head
Timm-based model with 5 separate heads for all targets (inference version).
EXP/EXP060/infer.py:915
↓ 1 callers
Class
TimmBiomassModel5HeadDensity
Timm-based model with Residual Density Head (inference version). Architecture: 1. DINOv3 ViT backbone: - CLS token → Glob
EXP/EXP113/infer.py:150
↓ 1 callers
Class
TimmBiomassModel5HeadGaussian
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 callers
Class
TimmBiomassModel5HeadHeightFiLM
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 callers
Class
TimmBiomassModel5HeadSoftSpecies
Timm-based model with Soft-Species Conditioning (inference version). Architecture: 1. timm Vision Model → global pooled embedding (e
EXP/EXP060/infer.py:985
↓ 1 callers
Class
TimmBiomassModelDensityOnly
Pure Density Integration Model (No Global Head) - Inference version. Architecture: 1. DINOv3 ViT backbone extracts patch tokens (32×
EXP/EXP113/infer.py:298
↓ 1 callers
Class
TimmBiomassModelDensityOnlyFast
Pure Density Integration Model optimized for fast inference. - Supports FP16 inference - Minimal overhead
EXP/EXP113/infer_fast.py:102
↓ 1 callers
Class
WeightedBiomassLoss
Weighted SmoothL1Loss for biomass prediction.
EXP/EXP060/train.py:2010
Class
BiomassDatasetOOF
Validation Dataset - returns both halves for OOF prediction.
EXP/EXP060/infer_ttt.py:151
Class
GradientReversalFunction
Gradient Reversal Layer (GRL) for Domain Adversarial Training. Forward pass: identity Backward pass: negate gradients and scale by lambd
EXP/EXP060/train.py:104
Class
GradientReversalFunction
Gradient Reversal Layer (GRL) for Domain Adversarial Training. Forward pass: identity Backward pass: negate gradients and scale by lambd
EXP/EXP113/train.py:78
Class
ResidualDilatedDensityHead
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
Class
ResidualDilatedDensityHead
Residual density head: stable 1×1 baseline + small dilated 3×3 context correction.
EXP/EXP113/infer.py:312
Class
SpatialDeepDensityHead
Spatial-Deep Density Head: DWConv for spatial context + MLP for non-linearity. Architecture: LayerNorm → 3x3 DWConv → Ba
EXP/EXP113/train.py:803
Class
SpatialDeepDensityHead
Spatial-Deep Density Head: DWConv for spatial context + MLP for non-linearity. Inference version (no initialization needed - weights
EXP/EXP113/infer.py:346