geowatch.tasks.invariants package

Subpackages

• geowatch.tasks.invariants.data package
  • Submodules
    • geowatch.tasks.invariants.data.datasets module
      • GriddedDataset
      • find_complete_image_indexes()
      • HashableBox
      • fixup_samples()
      • gridded_dataset
    • geowatch.tasks.invariants.data.multi_image_datasets module
      • kwcoco_dataset
      • SpaceNet7
    • geowatch.tasks.invariants.data.other_datasets module
      • kwcoco_dataset
      • Onera
      • SpaceNet7
  • Module contents
• geowatch.tasks.invariants.late_fusion package
  • Submodules
    • geowatch.tasks.invariants.late_fusion.fit_late_fusion module
      • main()
    • geowatch.tasks.invariants.late_fusion.pretext_model_late_fusion module
      • pretext
  • Module contents
• geowatch.tasks.invariants.utils package
  • Submodules
    • geowatch.tasks.invariants.utils.attention_unet module
      • positional_encoding
      • attention_unet
    • geowatch.tasks.invariants.utils.focal_loss module
      • FocalLoss
      • BinaryFocalLoss
    • geowatch.tasks.invariants.utils.read_sentinel_images module
      • adjust_shape()
      • read_sentinel_img()
      • read_sentinel_img_4()
      • read_sentinel_img_leq20()
      • read_sentinel_img_leq60()
      • read_sentinel_img_trio()
    • geowatch.tasks.invariants.utils.unet_blur module
      • BlurPool
      • get_pad_layer()
      • count_trainable_parameters()
      • double_conv
      • inconv
      • down
      • up
      • outconv
      • UNet_blur
      • UNetEncoder
      • UNetDecoder
    • geowatch.tasks.invariants.utils.unet_dropout module
      • BlurPool
      • get_pad_layer()
      • BlurPool1D
      • get_pad_layer_1d()
      • count_trainable_parameters()
      • double_conv
      • inconv
      • down
      • down_blur
      • up
      • outconv
      • UNet_Dropout
      • UNet_Blur_Dropout
      • UNet_Blur_Dropout_Encoder
  • Module contents

Submodules

• geowatch.tasks.invariants.change module
  • change
    • change.head()
    • change.forward()
    • change.shared_step()
    • change.training_step()
    • change.validation_step()
    • change.validation_epoch_end()
    • change.run_test()
    • change.train_dataloader()
    • change.val_dataloader()
    • change.configure_optimizers()
  • main()
• geowatch.tasks.invariants.fit module
  • main()
• geowatch.tasks.invariants.fit_segment module
  • main()
• geowatch.tasks.invariants.iarpa_dataset module
  • kwcoco_dataset
    • kwcoco_dataset.S2_channel_names
    • kwcoco_dataset.L8_channel_names
    • kwcoco_dataset.common_channel_names
    • kwcoco_dataset.get_img()
    • kwcoco_dataset.num_channels()
• geowatch.tasks.invariants.predict module
  • InvariantPredictConfig
    • InvariantPredictConfig.default
    • InvariantPredictConfig.normalize()
  • Predictor
    • Predictor.forward()
  • main()
• geowatch.tasks.invariants.pretext_model module
  • pretext
    • pretext.TASK_NAMES
    • pretext.forward()
    • pretext.shared_step()
    • pretext.training_step()
    • pretext.validation_step()
    • pretext.predict()
    • pretext.train_dataloader()
    • pretext.val_dataloader()
    • pretext.configure_optimizers()
    • pretext.task_neck()
    • pretext.pixel_classification_head()
    • pretext.image_classification_head()
    • pretext.generate_pca_matrix()
    • pretext.on_save_checkpoint()
    • pretext.save_package()
    • pretext.load_package()
• geowatch.tasks.invariants.segmentation_model module
  • segmentation_model
    • segmentation_model.forward()
    • segmentation_model.shared_step()
    • segmentation_model.training_step()
    • segmentation_model.validation_step()
    • segmentation_model.test_step()
    • segmentation_model.train_epoch_end()
    • segmentation_model.validation_epoch_end()
    • segmentation_model.run_test()
    • segmentation_model.train_dataloader()
    • segmentation_model.val_dataloader()
    • segmentation_model.configure_optimizers()
    • segmentation_model.save_package()
    • segmentation_model.load_package()

Module contents