geowatch.tasks.fusion package

Subpackages

• geowatch.tasks.fusion.architectures package
  • Submodules
    • geowatch.tasks.fusion.architectures.segmenter_decoder module
      • trunc_normal_()
      • drop_path()
      • DropPath
      • init_weights()
      • FeedForward
      • Attention
      • Block
      • DecoderLinear
      • MaskTransformerDecoder
    • geowatch.tasks.fusion.architectures.sits module
      • PositionalEncoding
      • BERTEmbedding
      • BERT
      • MulticlassClassification
      • BERTClassification
    • geowatch.tasks.fusion.architectures.transformer module
      • ResidualSequential
      • ResidualAttentionSequential
      • assert_allclose()
      • MultiheadSelfAttention
      • MetaModuleProperties
      • ModuleProperties
      • new_attention_layer()
      • new_mlp_layer()
      • ChannelwiseTransformerEncoderLayer
      • TimmEncoder
      • MM_VITEncoder
      • DeiTEncoder
      • PerceiverEncoder
      • FusionEncoder
      • default()
      • cache_fn()
      • PreNorm
      • GEGLU
      • FeedForward
      • Attention
      • BackboneEncoderDecoder
      • TransformerEncoderDecoder
      • TransformerEncoderLayerExtended
      • VanillaTransformerEncoder
      • MM_VITEncoderDecoder
    • geowatch.tasks.fusion.architectures.unet_blur module
      • BlurPool
      • get_pad_layer()
      • double_conv
      • inconv
      • down
      • up
      • outconv
      • UNet
    • geowatch.tasks.fusion.architectures.wu_mae module
      • PreNorm
      • FeedForward
      • Attention
      • pair()
      • Transformer
      • ViT
      • wu_backbone()
  • Module contents
• geowatch.tasks.fusion.datamodules package
  • Subpackages
    • geowatch.tasks.fusion.datamodules.temporal_sampling package
      • Submodules
      • Module contents
  • Submodules
    • geowatch.tasks.fusion.datamodules.batch_visualization module
      • BatchVisualizationBuilder
      • colorize_weights()
    • geowatch.tasks.fusion.datamodules.data_augment module
      • SpacetimeAugmentMixin
    • geowatch.tasks.fusion.datamodules.data_utils module
      • resolve_scale_request()
      • abslog_scaling()
      • fliprot()
      • fliprot_annot()
      • inv_fliprot_annot()
      • inv_fliprot()
      • samecolor_nodata_mask()
      • MultiscaleMask
    • geowatch.tasks.fusion.datamodules.kwcoco_datamodule module
      • KWCocoVideoDataModuleConfig
      • KWCocoVideoDataModule
    • geowatch.tasks.fusion.datamodules.kwcoco_dataset module
      • Known Issues
      • Roadmap
      • KWCocoVideoDatasetConfig
      • TruthMixin
      • GetItemMixin
      • IntrospectMixin
      • BalanceMixin
      • PreprocessMixin
      • MiscMixin
      • BackwardCompatMixin
      • KWCocoVideoDataset
      • more_demos()
      • worker_init_fn()
      • FailedSample
      • Modality
      • Domain
    • geowatch.tasks.fusion.datamodules.kwcoco_video_data module
    • geowatch.tasks.fusion.datamodules.network_io module
      • BatchItem
      • HeterogeneousBatchItem
      • HomogeneousBatchItem
      • RGBImageBatchItem
      • UncollatedBatch
      • HeterogeneousBatch
      • UncollatedRGBImageBatch
      • CollatedBatch
      • CollatedRGBImageBatch
      • NetworkOutputs
      • UncollatedNetworkOutputs
      • CollatedNetworkOutputs
      • decollate()
    • geowatch.tasks.fusion.datamodules.qa_bands module
      • QA_SpecMixin
      • QA_BitSpecTable
      • unpack_bit_positions()
      • QA_ValueSpecTable
      • QA_SpecRegistry
      • demo()
    • geowatch.tasks.fusion.datamodules.smart_mixins module
      • SMARTDataMixin
      • draw_cloudmask_viz()
    • geowatch.tasks.fusion.datamodules.spacetime_grid_builder module
      • SpacetimeGridBuilder
      • ImagePropertyCacher
      • sample_video_spacetime_targets()
  • Module contents
    • KWCocoVideoDataModule
      • KWCocoVideoDataModule.add_argparse_args()
      • KWCocoVideoDataModule.compatible()
      • KWCocoVideoDataModule.draw_batch()
      • KWCocoVideoDataModule.setup()
      • KWCocoVideoDataModule.test_dataset
      • KWCocoVideoDataModule.train_dataset
      • KWCocoVideoDataModule.vali_dataset
    • KWCocoVideoDataset
• geowatch.tasks.fusion.methods package
  • Submodules
    • geowatch.tasks.fusion.methods.channelwise_transformer module
      • MultimodalTransformerConfig
      • MultimodalTransformer
      • slice_to_agree()
      • perterb_params()
    • geowatch.tasks.fusion.methods.efficientdet module
      • normal_init()
      • bias_init_with_prob()
      • conv_ws_2d()
      • ConvWS2d
      • build_norm_layer()
      • build_conv_layer()
      • ConvModule
      • multi_apply()
      • RetinaHead
      • calc_iou()
      • FocalLoss
      • drop_connect()
      • round_repeats()
      • round_filters()
      • load_pretrained_weights()
      • Identity
      • Conv2dStaticSamePadding
      • Conv2dDynamicSamePadding
      • get_same_padding_conv2d()
      • BlockArgs
      • GlobalParams
      • BlockDecoder
      • efficientnet_params()
      • efficientnet()
      • get_model_params()
      • Swish
      • SwishImplementation
      • MemoryEfficientSwish
      • MBConvBlock
      • EfficientNet
      • ClipBoxes
      • xavier_init()
      • BiFPNModule
      • BIFPN
      • BBoxTransform
      • shift()
      • generate_anchors()
      • Anchors
      • EfficientDetCoder
      • EfficientDet
    • geowatch.tasks.fusion.methods.heads module
      • TaskHeads
    • geowatch.tasks.fusion.methods.heterogeneous module
      • to_next_multiple()
      • positions_from_shape()
      • PadToMultiple
      • NanToNum
      • ShapePreservingTransformerEncoder
      • ScaleAwarePositionalEncoder
      • MipNerfPositionalEncoder
      • ScaleAgnostictPositionalEncoder
      • ResNetShim
      • HeterogeneousModel
    • geowatch.tasks.fusion.methods.loss module
      • coerce_criterion()
    • geowatch.tasks.fusion.methods.network_modules module
      • drop_path()
      • RobustModuleDict
      • RobustParameterDict
      • OurDepthwiseSeparableConv
      • DWCNNTokenizer
      • LinearConvTokenizer
      • ConvTokenizer
      • RearrangeTokenizer
      • torch_safe_stack()
    • geowatch.tasks.fusion.methods.noop_model module
      • NoopModel
    • geowatch.tasks.fusion.methods.object_head module
      • DetrDecoderForObjectDetection
    • geowatch.tasks.fusion.methods.torchvision_nets module
      • TorchvisionWrapper
      • TorchvisionSegmentationWrapper
      • TorchvisionClassificationWrapper
      • TorchvisionDetectionWrapper
      • EfficientNetB7
      • FCNResNet50
      • Resnet50
    • geowatch.tasks.fusion.methods.unet_baseline module
      • NanToNum
      • UNetBaseline
    • geowatch.tasks.fusion.methods.watch_module_mixins module
      • ExtendTorchMixin
      • MSIDemoDataMixin
      • LightningModelMixin
      • DeprecatedMixin
      • OverfitMixin
      • PackageMixin
      • CoerceMixins
      • DatasetStatsMixin
      • WatchModuleMixins
  • Module contents
    • MultimodalTransformer
      • MultimodalTransformer.add_argparse_args()
      • MultimodalTransformer.compatible()
      • MultimodalTransformer.configure_optimizers()
      • MultimodalTransformer.forward()
      • MultimodalTransformer.forward_foot()
      • MultimodalTransformer.forward_item()
      • MultimodalTransformer.forward_step()
      • MultimodalTransformer.get_cfgstr()
      • MultimodalTransformer.optimizer_step()
      • MultimodalTransformer.overfit()
      • MultimodalTransformer.parameter_hacking()
      • MultimodalTransformer.prepare_item()
      • MultimodalTransformer.save_package()
      • MultimodalTransformer.test_step()
      • MultimodalTransformer.training_step()
      • MultimodalTransformer.validation_step()
    • HeterogeneousModel
      • HeterogeneousModel.configure_optimizers()
      • HeterogeneousModel.forward()
      • HeterogeneousModel.forward_step()
      • HeterogeneousModel.get_cfgstr()
      • HeterogeneousModel.log_grad_norm()
      • HeterogeneousModel.predict_step()
      • HeterogeneousModel.process_input_tokens()
      • HeterogeneousModel.process_query_tokens()
      • HeterogeneousModel.save_package()
      • HeterogeneousModel.shared_step()
      • HeterogeneousModel.test_step()
      • HeterogeneousModel.training_step()
      • HeterogeneousModel.validation_step()
    • UNetBaseline
      • UNetBaseline.encode_batch()
      • UNetBaseline.encode_example()
      • UNetBaseline.encode_frame()
      • UNetBaseline.forward()
      • UNetBaseline.forward_step()
      • UNetBaseline.get_cfgstr()
      • UNetBaseline.predict_step()
      • UNetBaseline.process_batch()
      • UNetBaseline.process_example()
      • UNetBaseline.process_frame()
      • UNetBaseline.save_package()
      • UNetBaseline.shared_step()
      • UNetBaseline.test_step()
      • UNetBaseline.training_step()
      • UNetBaseline.validation_step()
    • NoopModel
      • NoopModel.configure_optimizers()
      • NoopModel.forward()
      • NoopModel.forward_step()
      • NoopModel.get_cfgstr()
      • NoopModel.save_package()
      • NoopModel.shared_step()
      • NoopModel.training_step()

Submodules

• geowatch.tasks.fusion.coco_stitcher module
  • demo_coco_stitching_manager()
  • CocoStitchingManager
    • CocoStitchingManager.accumulate_image()
    • CocoStitchingManager.managed_image_ids()
    • CocoStitchingManager.ready_image_ids()
    • CocoStitchingManager.submit_finalize_image()
    • CocoStitchingManager.flush_images()
    • CocoStitchingManager.seen_image_ids
    • CocoStitchingManager.finalize_image()
  • quantize_image()
  • fix_slice()
• geowatch.tasks.fusion.evaluate module
  • SegmentationEvalConfig
    • SegmentationEvalConfig.default
  • main()
  • single_image_segmentation_metrics()
  • draw_confusion_image()
  • colorize_class_probs()
  • draw_truth_borders()
  • dump_chunked_confusion()
  • evaluate_segmentations()
  • MaxQueuePool
    • MaxQueuePool.submit()
    • MaxQueuePool.pool_queue_callback()
    • MaxQueuePool.shutdown()
• geowatch.tasks.fusion.fit module
  • coerce_initializer()
• geowatch.tasks.fusion.fit_lightning module
  • custom_yaml_load()
  • custom_yaml_dump()
  • SmartTrainer
    • SmartTrainer.log_dpath
  • TorchGlobals
    • TorchGlobals.setup()
  • WeightInitializer
    • WeightInitializer.setup()
  • SmartLightningCLI
    • SmartLightningCLI.configure_optimizers()
    • SmartLightningCLI.add_arguments_to_parser()
  • instantiate_datamodule()
  • make_cli()
  • main()
• geowatch.tasks.fusion.fit_registery module
  • Usage
  • Registery
    • Registery.pandas()
    • Registery.list()
    • Registery.query()
  • DictRegistery
    • DictRegistery.add()
    • DictRegistery.set()
    • DictRegistery.remove()
    • DictRegistery.read()
  • ListRegistery
    • ListRegistery.append()
    • ListRegistery.read()
    • ListRegistery.pandas()
  • NewDVCRegistery
  • FitRegistery
    • FitRegistery.append()
    • FitRegistery.peek()
  • FitRegisteryCLI
    • FitRegisteryCLI.Append
      • FitRegisteryCLI.Append.main()
      • FitRegisteryCLI.Append.default
    • FitRegisteryCLI.List
      • FitRegisteryCLI.List.main()
      • FitRegisteryCLI.List.default
    • FitRegisteryCLI.Peek
      • FitRegisteryCLI.Peek.main()
      • FitRegisteryCLI.Peek.default
• geowatch.tasks.fusion.helper_scripts module
  • train_time_helper_scripts()
• geowatch.tasks.fusion.predict module
  • DataModuleConfigMixin
    • DataModuleConfigMixin.default
  • PredictConfig
    • PredictConfig.default
  • build_stitching_managers()
  • resolve_datamodule()
  • PeriodicMemoryMonitor
    • PeriodicMemoryMonitor.check()
  • predict()
  • Predictor
  • main()
• geowatch.tasks.fusion.production module
• geowatch.tasks.fusion.utils module
  • millify()
  • load_model_from_package()
  • load_model_header()
  • Lambda
    • Lambda.forward()
  • DimensionDropout
    • DimensionDropout.forward()
  • SinePositionalEncoding
    • SinePositionalEncoding.forward()
  • model_json()

Module contents

mkinit geowatch.tasks.fusion –lazy –noattrs -w mkinit -m geowatch.tasks.fusion –noattrs -w