geowatch.tasks.fusion package

Subpackages

• geowatch.tasks.fusion.architectures package
  • Submodules
    • geowatch.tasks.fusion.architectures.segmenter_decoder module
      • _no_grad_trunc_normal_()
      • trunc_normal_()
      • drop_path()
      • DropPath
      • init_weights()
      • FeedForward
      • Attention
      • Block
      • DecoderLinear
      • _string_to_hashvec()
      • MaskTransformerDecoder
    • geowatch.tasks.fusion.architectures.sits module
      • PositionalEncoding
      • BERTEmbedding
      • BERT
      • MulticlassClassification
      • BERTClassification
    • geowatch.tasks.fusion.architectures.transformer module
      • ResidualSequential
      • ResidualAttentionSequential
      • assert_allclose()
      • MultiheadSelfAttention
      • new_attention_layer()
      • new_mlp_layer()
      • ChannelwiseTransformerEncoderLayer
      • TimmEncoder
      • MM_VITEncoder
      • DeiTEncoder
      • PerceiverEncoder
      • FusionEncoder
      • _build_global_configs()
      • 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
      • _draw_overlay_item_by_itself()
      • _draw_row_item()
      • _debug_sample_in_context()
      • colorize_weights()
    • geowatch.tasks.fusion.datamodules.data_augment module
      • SpacetimeAugmentMixin
    • geowatch.tasks.fusion.datamodules.data_utils module
      • resolve_scale_request()
      • polygon_distance_transform()
      • abslog_scaling()
      • fliprot()
      • fliprot_annot()
      • inv_fliprot_annot()
      • inv_fliprot()
      • _string_to_hashvec()
      • _boxes_snap_to_edges()
      • NestedPool
      • samecolor_nodata_mask()
      • MultiscaleMask
    • geowatch.tasks.fusion.datamodules.kwcoco_datamodule module
      • KWCocoVideoDataModuleConfig
      • KWCocoVideoDataModule
    • geowatch.tasks.fusion.datamodules.kwcoco_dataset module
      • Known Issues
      • KWCocoVideoDatasetConfig
      • TruthMixin
      • GetItemMixin
      • IntrospectMixin
      • BalanceMixin
      • PreprocessMixin
      • MiscMixin
      • KWCocoVideoDataset
      • more_demos()
      • worker_init_fn()
      • _space_weights()
      • FailedSample
      • Modality
      • Domain
    • geowatch.tasks.fusion.datamodules.kwcoco_video_data module
    • geowatch.tasks.fusion.datamodules.qa_bands module
      • _dump_qa_debug_vid()
      • 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
      • sample_video_spacetime_targets()
      • ImagePropertyCacher
      • _sample_single_video_spacetime_targets()
      • _build_targets_around_track()
      • _build_targets_in_spatial_region()
      • _build_vidspace_track_qtree()
      • _refine_time_sample()
      • visualize_sample_grid()
  • Module contents
    • KWCocoVideoDataModule
      • KWCocoVideoDataModule.setup()
      • KWCocoVideoDataModule.train_dataset
      • KWCocoVideoDataModule.test_dataset
      • KWCocoVideoDataModule.vali_dataset
      • KWCocoVideoDataModule._make_dataloader()
      • KWCocoVideoDataModule._notify_about_tasks()
      • KWCocoVideoDataModule.add_argparse_args()
      • KWCocoVideoDataModule.compatible()
      • KWCocoVideoDataModule.draw_batch()
    • 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.heterogeneous module
      • to_next_multiple()
      • positions_from_shape()
      • PadToMultiple
      • NanToNum
      • ShapePreservingTransformerEncoder
      • ScaleAwarePositionalEncoder
      • MipNerfPositionalEncoder
      • ScaleAgnostictPositionalEncoder
      • ResNetShim
      • HeterogeneousModel
    • geowatch.tasks.fusion.methods.network_modules module
      • drop_path()
      • RobustModuleDict
      • RobustParameterDict
      • OurDepthwiseSeparableConv
      • DWCNNTokenizer
      • LinearConvTokenizer
      • ConvTokenizer
      • RearrangeTokenizer
      • _torch_meshgrid()
      • _class_weights_from_freq()
      • coerce_criterion()
      • torch_safe_stack()
    • geowatch.tasks.fusion.methods.noop_model module
      • NoopModel
    • geowatch.tasks.fusion.methods.unet_baseline module
      • NanToNum
      • UNetBaseline
    • geowatch.tasks.fusion.methods.watch_module_mixins module
      • WatchModuleMixins
  • Module contents
    • MultimodalTransformer
      • MultimodalTransformer._HANDLES_NANS
      • MultimodalTransformer.get_cfgstr()
      • MultimodalTransformer.add_argparse_args()
      • MultimodalTransformer.compatible()
      • MultimodalTransformer.configure_optimizers()
      • MultimodalTransformer.overfit()
      • MultimodalTransformer.prepare_item()
      • MultimodalTransformer.forward_step()
      • MultimodalTransformer.forward_item()
      • MultimodalTransformer.forward_foot()
      • MultimodalTransformer._head_loss()
      • MultimodalTransformer._build_item_loss_parts()
      • MultimodalTransformer.training_step()
      • MultimodalTransformer.optimizer_step()
      • MultimodalTransformer.parameter_hacking()
      • MultimodalTransformer.validation_step()
      • MultimodalTransformer.test_step()
      • MultimodalTransformer.save_package()
      • MultimodalTransformer.forward()
    • HeterogeneousModel
      • HeterogeneousModel._HANDLES_NANS
      • HeterogeneousModel.get_cfgstr()
      • HeterogeneousModel.process_input_tokens()
      • HeterogeneousModel.process_query_tokens()
      • HeterogeneousModel.forward()
      • HeterogeneousModel.shared_step()
      • HeterogeneousModel.training_step()
      • HeterogeneousModel.validation_step()
      • HeterogeneousModel.test_step()
      • HeterogeneousModel.predict_step()
      • HeterogeneousModel.forward_step()
      • HeterogeneousModel.log_grad_norm()
      • HeterogeneousModel.save_package()
      • HeterogeneousModel.configure_optimizers()
    • UNetBaseline
      • UNetBaseline._HANDLES_NANS
      • UNetBaseline.get_cfgstr()
      • UNetBaseline.process_frame()
      • UNetBaseline.process_example()
      • UNetBaseline.process_batch()
      • UNetBaseline.encode_frame()
      • UNetBaseline.encode_example()
      • UNetBaseline.encode_batch()
      • UNetBaseline.forward()
      • UNetBaseline.shared_step()
      • UNetBaseline.training_step()
      • UNetBaseline.validation_step()
      • UNetBaseline.test_step()
      • UNetBaseline.predict_step()
      • UNetBaseline.forward_step()
      • UNetBaseline.save_package()
    • NoopModel
      • NoopModel._HANDLES_NANS
      • NoopModel.get_cfgstr()
      • NoopModel.forward()
      • NoopModel.shared_step()
      • NoopModel.training_step()
      • NoopModel.forward_step()
      • NoopModel.configure_optimizers()
      • NoopModel.save_package()

Submodules

• geowatch.tasks.fusion.__main__ module
• geowatch.tasks.fusion.coco_stitcher module
  • demo_coco_stitching_manager()
  • CocoStitchingManager
    • CocoStitchingManager.accumulate_image()
    • CocoStitchingManager._stitcher_center_weighted_add()
    • CocoStitchingManager.managed_image_ids()
    • CocoStitchingManager.ready_image_ids()
    • CocoStitchingManager.submit_finalize_image()
    • CocoStitchingManager.flush_images()
    • CocoStitchingManager.seen_image_ids
    • CocoStitchingManager.finalize_image()
  • quantize_image()
  • quantize_float01()
  • fix_slice()
  • _fix_int()
  • _fix_slice()
  • _fix_slice_tup()
  • _force_shape_agreement_by_cropping2d()
• geowatch.tasks.fusion.evaluate module
  • SegmentationEvalConfig
    • SegmentationEvalConfig.default
  • main()
  • single_image_segmentation_metrics()
  • _memo_legend()
  • draw_confusion_image()
  • colorize_class_probs()
  • draw_truth_borders()
  • dump_chunked_confusion()
  • evaluate_segmentations()
  • _redraw_measures()
• geowatch.tasks.fusion.fit module
  • coerce_initializer()
• geowatch.tasks.fusion.fit_lightning module
  • custom_yaml_load()
  • custom_yaml_dump()
  • SmartTrainer
    • SmartTrainer._run_stage()
    • SmartTrainer._write_inspect_helper_scripts()
  • TorchGlobals
    • TorchGlobals.setup()
  • WeightInitializer
    • WeightInitializer.setup()
  • SmartLightningCLI
    • SmartLightningCLI.configure_optimizers()
    • SmartLightningCLI.add_arguments_to_parser()
  • instantiate_datamodule()
  • _final_pkg_compute_fn()
  • _ValueGetter
  • _data_value_getter()
  • make_cli()
  • main()
• geowatch.tasks.fusion.predict module
  • DataModuleConfigMixin
    • DataModuleConfigMixin.default
  • PredictConfig
    • PredictConfig.default
  • build_stitching_managers()
  • resolve_datamodule()
  • _prepare_predict_data()
  • _debug_grid()
  • _jsonify()
  • _predict_critical_loop()
  • predict()
  • main()
• geowatch.tasks.fusion.production module
• geowatch.tasks.fusion.utils module
  • millify()
  • _map_location()
  • load_model_from_package()
  • _try_fixed_package_import()
  • Lambda
    • Lambda.forward()
  • DimensionDropout
    • DimensionDropout.forward()
  • ordinal_position_encoding()
  • SinePositionalEncoding
    • SinePositionalEncoding._encoding_part()
    • SinePositionalEncoding.forward()
  • model_json()
  • _memo_legend()
  • category_tree_ensure_color()

Module contents

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