API documentation

Base

• Base
  • aitlas.base.classification module
    • BaseMulticlassClassifier
      • BaseMulticlassClassifier.schema
      • BaseMulticlassClassifier.get_predicted()
      • BaseMulticlassClassifier.report()
      • BaseMulticlassClassifier.load_optimizer()
      • BaseMulticlassClassifier.load_criterion()
      • BaseMulticlassClassifier.load_lr_scheduler()
      • BaseMulticlassClassifier.training
    • BaseMultilabelClassifier
      • BaseMultilabelClassifier.schema
      • BaseMultilabelClassifier.load_optimizer()
      • BaseMultilabelClassifier.load_criterion()
      • BaseMultilabelClassifier.load_lr_scheduler()
      • BaseMultilabelClassifier.get_predicted()
      • BaseMultilabelClassifier.report()
      • BaseMultilabelClassifier.training
  • aitlas.base.config module
    • Config
    • ObjectConfig
      • ObjectConfig.opts
    • RunConfig
      • RunConfig.opts
    • Configurable
      • Configurable.schema
  • aitlas.base.datasets module
    • BaseDataset
      • BaseDataset.schema
      • BaseDataset.name
      • BaseDataset.labels
      • BaseDataset.get_name()
      • BaseDataset.prepare()
      • BaseDataset.dataloader()
      • BaseDataset.get_labels()
      • BaseDataset.show_batch()
      • BaseDataset.show_samples()
      • BaseDataset.show_image()
      • BaseDataset.data_distribution_table()
      • BaseDataset.data_distribution_barchart()
      • BaseDataset.load_transforms()
  • aitlas.base.metrics module
    • BaseMetric
      • BaseMetric.calculate()
    • RunningScore
      • RunningScore.update()
      • RunningScore.reset()
      • RunningScore.get_computed()
      • RunningScore.precision()
      • RunningScore.accuracy()
      • RunningScore.weights()
      • RunningScore.recall()
      • RunningScore.f1_score()
      • RunningScore.iou()
      • RunningScore.get_scores()
    • MultiClassRunningScore
      • MultiClassRunningScore.accuracy()
      • MultiClassRunningScore.weights()
      • MultiClassRunningScore.recall()
      • MultiClassRunningScore.precision()
      • MultiClassRunningScore.iou()
      • MultiClassRunningScore.kappa()
    • MultiLabelRunningScore
      • MultiLabelRunningScore.reset()
      • MultiLabelRunningScore.update()
      • MultiLabelRunningScore.map()
      • MultiLabelRunningScore.roc_auc_score()
      • MultiLabelRunningScore.accuracy()
      • MultiLabelRunningScore.precision()
      • MultiLabelRunningScore.weights()
      • MultiLabelRunningScore.recall()
      • MultiLabelRunningScore.get_outcomes()
      • MultiLabelRunningScore.count()
      • MultiLabelRunningScore.get_samples()
      • MultiLabelRunningScore.iou()
    • SegmentationRunningScore
      • SegmentationRunningScore.update()
    • ObjectDetectionRunningScore
      • ObjectDetectionRunningScore.update()
      • ObjectDetectionRunningScore.reset()
      • ObjectDetectionRunningScore.compute()
      • ObjectDetectionRunningScore.map()
      • ObjectDetectionRunningScore.map_50()
      • ObjectDetectionRunningScore.get_scores()
  • aitlas.base.models module
    • EarlyStopping
    • BaseModel
      • BaseModel.schema
      • BaseModel.name
      • BaseModel.log_loss
      • BaseModel.prepare()
      • BaseModel.fit()
      • BaseModel.train_epoch()
      • BaseModel.evaluate()
      • BaseModel.evaluate_model()
      • BaseModel.predict()
      • BaseModel.predict_image()
      • BaseModel.predict_masks()
      • BaseModel.detect_objects()
      • BaseModel.predict_output_per_batch()
      • BaseModel.forward()
      • BaseModel.get_predicted()
      • BaseModel.report()
      • BaseModel.log_metrics()
      • BaseModel.allocate_device()
      • BaseModel.save_model()
      • BaseModel.extract_features()
      • BaseModel.load_model()
      • BaseModel.load_optimizer()
      • BaseModel.load_criterion()
      • BaseModel.load_lr_scheduler()
      • BaseModel.train_model()
      • BaseModel.train_and_evaluate_model()
      • BaseModel.training
  • aitlas.base.object_detection module
    • BaseObjectDetection
      • BaseObjectDetection.schema
      • BaseObjectDetection.log_loss
      • BaseObjectDetection.get_predicted()
      • BaseObjectDetection.load_optimizer()
      • BaseObjectDetection.load_criterion()
      • BaseObjectDetection.load_lr_scheduler()
      • BaseObjectDetection.train_epoch()
      • BaseObjectDetection.predict_output_per_batch()
      • BaseObjectDetection.evaluate_model()
      • BaseObjectDetection.training
  • aitlas.base.schemas module
    • BaseDatasetSchema
      • BaseDatasetSchema.opts
    • BaseModelSchema
      • BaseModelSchema.opts
    • BaseClassifierSchema
      • BaseClassifierSchema.opts
      • BaseClassifierSchema.fields
      • BaseClassifierSchema.load_fields
      • BaseClassifierSchema.dump_fields
    • BaseSegmentationClassifierSchema
      • BaseSegmentationClassifierSchema.opts
      • BaseSegmentationClassifierSchema.fields
      • BaseSegmentationClassifierSchema.load_fields
      • BaseSegmentationClassifierSchema.dump_fields
    • BaseObjectDetectionSchema
      • BaseObjectDetectionSchema.opts
      • BaseObjectDetectionSchema.fields
      • BaseObjectDetectionSchema.load_fields
      • BaseObjectDetectionSchema.dump_fields
    • BaseTransformsSchema
      • BaseTransformsSchema.opts
  • aitlas.base.segmentation module
    • BaseSegmentationClassifier
      • BaseSegmentationClassifier.schema
      • BaseSegmentationClassifier.get_predicted()
      • BaseSegmentationClassifier.load_optimizer()
      • BaseSegmentationClassifier.load_criterion()
      • BaseSegmentationClassifier.load_lr_scheduler()
      • BaseSegmentationClassifier.training
  • aitlas.base.tasks module
    • BaseTask
      • BaseTask.create_dataset()
      • BaseTask.generate_task_id()
      • BaseTask.run()
  • aitlas.base.transforms module
    • load_transforms()
    • BaseTransforms
      • BaseTransforms.schema
      • BaseTransforms.configurables
  • aitlas.base.visualizations module
    • BaseVisualization
      • BaseVisualization.plot()
    • BaseDetailedVisualization
      • BaseDetailedVisualization.plot()

Datasets

• Datasets
  • aitlas.datasets.aid module
    • AIDDataset
      • AIDDataset.url
      • AIDDataset.labels
      • AIDDataset.name
  • aitlas.datasets.aid_multilabel module
    • AIDMultiLabelDataset
      • AIDMultiLabelDataset.url
      • AIDMultiLabelDataset.labels
      • AIDMultiLabelDataset.name
  • aitlas.datasets.airs module
    • AIRSDataset
      • AIRSDataset.url
      • AIRSDataset.labels
      • AIRSDataset.color_mapping
      • AIRSDataset.name
  • aitlas.datasets.amazon_rainforest module
    • AmazonRainforestDataset
      • AmazonRainforestDataset.url
      • AmazonRainforestDataset.labels
      • AmazonRainforestDataset.color_mapping
      • AmazonRainforestDataset.name
      • AmazonRainforestDataset.load_dataset()
  • aitlas.datasets.big_earth_net module
    • interp_band()
    • parse_json_labels()
    • update_json_labels()
    • loads_pickle()
    • dumps_pickle()
    • cls2multihot()
    • BigEarthNetDataset
      • BigEarthNetDataset.schema
      • BigEarthNetDataset.name
      • BigEarthNetDataset.get_labels()
      • BigEarthNetDataset.load_patches()
      • BigEarthNetDataset.get_item_name()
      • BigEarthNetDataset.show_image()
      • BigEarthNetDataset.save_image()
      • BigEarthNetDataset.show_batch()
      • BigEarthNetDataset.data_distribution_table()
      • BigEarthNetDataset.data_distribution_barchart()
      • BigEarthNetDataset.labels_stats()
      • BigEarthNetDataset.prepare()
      • BigEarthNetDataset.process_to_lmdb()
    • PrepBigEarthNetDataset
  • aitlas.datasets.brazilian_coffee_scenes module
    • BrazilianCoffeeScenesDataset
      • BrazilianCoffeeScenesDataset.url
      • BrazilianCoffeeScenesDataset.labels
      • BrazilianCoffeeScenesDataset.name
    • prepare()
  • aitlas.datasets.breizhcrops module
    • DownloadProgressBar
      • DownloadProgressBar.update_to()
    • download_file()
    • unzip()
    • untar()
    • BreizhCropsDataset
      • BreizhCropsDataset.schema
      • BreizhCropsDataset.preprocess()
      • BreizhCropsDataset.get_labels()
      • BreizhCropsDataset.data_distribution_table()
      • BreizhCropsDataset.parcel_distribution_table()
      • BreizhCropsDataset.data_distribution_barchart()
      • BreizhCropsDataset.show_samples()
      • BreizhCropsDataset.show_timeseries()
      • BreizhCropsDataset.download_csv_files()
      • BreizhCropsDataset.build_folder_structure()
      • BreizhCropsDataset.get_fid()
      • BreizhCropsDataset.download_h5_database()
      • BreizhCropsDataset.write_h5_database_from_csv()
      • BreizhCropsDataset.get_codes()
      • BreizhCropsDataset.load_classmapping()
      • BreizhCropsDataset.get_classes_to_ind()
      • BreizhCropsDataset.load_raw()
      • BreizhCropsDataset.load()
      • BreizhCropsDataset.load_culturecode_and_id()
      • BreizhCropsDataset.write_index()
  • aitlas.datasets.camvid module
    • CamVidDataset
      • CamVidDataset.url
      • CamVidDataset.labels
      • CamVidDataset.color_mapping
      • CamVidDataset.name
      • CamVidDataset.load_dataset()
  • aitlas.datasets.chactun module
    • ChactunDataset
      • ChactunDataset.labels
      • ChactunDataset.color_mapping
      • ChactunDataset.name
      • ChactunDataset.load_dataset()
      • ChactunDataset.show_image()
  • aitlas.datasets.clrs module
    • CLRSDataset
      • CLRSDataset.url
      • CLRSDataset.labels
      • CLRSDataset.name
  • aitlas.datasets.crops_classification module
    • CropsDataset
      • CropsDataset.schema
      • CropsDataset.preprocess()
      • CropsDataset.get_labels()
      • CropsDataset.data_distribution_table()
      • CropsDataset.parcel_distribution_table()
      • CropsDataset.data_distribution_barchart()
      • CropsDataset.show_samples()
      • CropsDataset.show_image()
      • CropsDataset.show_timeseries()
      • CropsDataset.get_codes()
      • CropsDataset.load_classmapping()
  • aitlas.datasets.dfc15_multilabel module
    • DFC15MultiLabelDataset
      • DFC15MultiLabelDataset.url
      • DFC15MultiLabelDataset.labels
      • DFC15MultiLabelDataset.name
  • aitlas.datasets.eopatch_crops module
    • DownloadProgressBar
      • DownloadProgressBar.update_to()
    • download_file()
    • EOPatchCrops
      • EOPatchCrops.preprocess()
      • EOPatchCrops.split()
      • EOPatchCrops.write_index()
  • aitlas.datasets.eurosat module
    • EurosatDataset
      • EurosatDataset.url
      • EurosatDataset.labels
      • EurosatDataset.name
  • aitlas.datasets.inria module
    • InriaDataset
      • InriaDataset.url
      • InriaDataset.labels
      • InriaDataset.color_mapping
      • InriaDataset.name
  • aitlas.datasets.landcover_ai module
    • LandCoverAiDataset
      • LandCoverAiDataset.url
      • LandCoverAiDataset.labels
      • LandCoverAiDataset.color_mapping
      • LandCoverAiDataset.name
    • split_images()
  • aitlas.datasets.massachusetts_buildings module
    • MassachusettsBuildingsDataset
      • MassachusettsBuildingsDataset.url
      • MassachusettsBuildingsDataset.labels
      • MassachusettsBuildingsDataset.color_mapping
      • MassachusettsBuildingsDataset.name
  • aitlas.datasets.massachusetts_roads module
    • MassachusettsRoadsDataset
      • MassachusettsRoadsDataset.url
      • MassachusettsRoadsDataset.labels
      • MassachusettsRoadsDataset.color_mapping
      • MassachusettsRoadsDataset.name
  • aitlas.datasets.mlrs_net module
    • MLRSNetMultiLabelDataset
      • MLRSNetMultiLabelDataset.url
      • MLRSNetMultiLabelDataset.labels
      • MLRSNetMultiLabelDataset.name
    • prepare()
  • aitlas.datasets.multiclass_classification module
    • MultiClassClassificationDataset
      • MultiClassClassificationDataset.schema
      • MultiClassClassificationDataset.get_labels()
      • MultiClassClassificationDataset.data_distribution_table()
      • MultiClassClassificationDataset.data_distribution_barchart()
      • MultiClassClassificationDataset.show_samples()
      • MultiClassClassificationDataset.show_image()
      • MultiClassClassificationDataset.show_batch()
      • MultiClassClassificationDataset.load_dataset()
      • MultiClassClassificationDataset.re_map_labels()
  • aitlas.datasets.multilabel_classification module
    • MultiLabelClassificationDataset
      • MultiLabelClassificationDataset.schema
      • MultiLabelClassificationDataset.get_labels()
      • MultiLabelClassificationDataset.data_distribution_table()
      • MultiLabelClassificationDataset.data_distribution_barchart()
      • MultiLabelClassificationDataset.show_samples()
      • MultiLabelClassificationDataset.show_image()
      • MultiLabelClassificationDataset.show_batch()
      • MultiLabelClassificationDataset.load_dataset()
      • MultiLabelClassificationDataset.labels_stats()
      • MultiLabelClassificationDataset.re_map_labels()
  • aitlas.datasets.npz module
    • NpzDataset
      • NpzDataset.schema
      • NpzDataset.labels
      • NpzDataset.get_labels()
      • NpzDataset.data_distribution_table()
      • NpzDataset.data_distribution_barchart()
      • NpzDataset.show_samples()
      • NpzDataset.show_image()
      • NpzDataset.show_batch()
      • NpzDataset.load_dataset()
  • aitlas.datasets.object_detection module
    • BaseObjectDetectionDataset
      • BaseObjectDetectionDataset.name
      • BaseObjectDetectionDataset.dataloader()
      • BaseObjectDetectionDataset.apply_transformations()
      • BaseObjectDetectionDataset.get_labels()
      • BaseObjectDetectionDataset.show_image()
      • BaseObjectDetectionDataset.show_batch()
    • ObjectDetectionPascalDataset
      • ObjectDetectionPascalDataset.schema
      • ObjectDetectionPascalDataset.labels
      • ObjectDetectionPascalDataset.load_dataset()
      • ObjectDetectionPascalDataset.data_distribution_table()
      • ObjectDetectionPascalDataset.data_distribution_barchart()
    • ObjectDetectionCocoDataset
      • ObjectDetectionCocoDataset.schema
      • ObjectDetectionCocoDataset.data_distribution_table()
      • ObjectDetectionCocoDataset.data_distribution_barchart()
      • ObjectDetectionCocoDataset.show_samples()
      • ObjectDetectionCocoDataset.load_dataset()
  • aitlas.datasets.optimal_31 module
    • Optimal31Dataset
      • Optimal31Dataset.url
      • Optimal31Dataset.labels
      • Optimal31Dataset.name
  • aitlas.datasets.pattern_net module
    • PatternNetDataset
      • PatternNetDataset.url
      • PatternNetDataset.labels
      • PatternNetDataset.name
  • aitlas.datasets.planet_uas module
    • PlanetUASMultiLabelDataset
      • PlanetUASMultiLabelDataset.url
      • PlanetUASMultiLabelDataset.labels
      • PlanetUASMultiLabelDataset.name
    • prepare()
    • kaggle_format()
  • aitlas.datasets.resisc45 module
    • Resisc45Dataset
      • Resisc45Dataset.url
      • Resisc45Dataset.labels
      • Resisc45Dataset.name
  • aitlas.datasets.rsd46_whu module
    • RSD46WHUDataset
      • RSD46WHUDataset.url
      • RSD46WHUDataset.labels
      • RSD46WHUDataset.name
  • aitlas.datasets.rsi_cb256 module
    • RSICB256Dataset
      • RSICB256Dataset.url
      • RSICB256Dataset.labels
      • RSICB256Dataset.name
  • aitlas.datasets.rsscn7 module
    • RSSCN7Dataset
      • RSSCN7Dataset.url
      • RSSCN7Dataset.labels
      • RSSCN7Dataset.name
  • aitlas.datasets.sat6 module
    • SAT6Dataset
      • SAT6Dataset.schema
      • SAT6Dataset.url
      • SAT6Dataset.labels
      • SAT6Dataset.name
      • SAT6Dataset.get_labels()
      • SAT6Dataset.data_distribution_table()
      • SAT6Dataset.data_distribution_barchart()
      • SAT6Dataset.show_image()
      • SAT6Dataset.show_batch()
      • SAT6Dataset.load_dataset()
      • SAT6Dataset.re_map_labels()
  • aitlas.datasets.schemas module
    • MatDatasetSchema
      • MatDatasetSchema.opts
      • MatDatasetSchema.fields
      • MatDatasetSchema.load_fields
      • MatDatasetSchema.dump_fields
    • NPZDatasetSchema
      • NPZDatasetSchema.opts
      • NPZDatasetSchema.fields
      • NPZDatasetSchema.load_fields
      • NPZDatasetSchema.dump_fields
    • ClassificationDatasetSchema
      • ClassificationDatasetSchema.opts
      • ClassificationDatasetSchema.fields
      • ClassificationDatasetSchema.load_fields
      • ClassificationDatasetSchema.dump_fields
    • SegmentationDatasetSchema
      • SegmentationDatasetSchema.opts
      • SegmentationDatasetSchema.fields
      • SegmentationDatasetSchema.load_fields
      • SegmentationDatasetSchema.dump_fields
    • ObjectDetectionPascalDatasetSchema
      • ObjectDetectionPascalDatasetSchema.opts
      • ObjectDetectionPascalDatasetSchema.fields
      • ObjectDetectionPascalDatasetSchema.load_fields
      • ObjectDetectionPascalDatasetSchema.dump_fields
    • ObjectDetectionCocoDatasetSchema
      • ObjectDetectionCocoDatasetSchema.opts
      • ObjectDetectionCocoDatasetSchema.fields
      • ObjectDetectionCocoDatasetSchema.load_fields
      • ObjectDetectionCocoDatasetSchema.dump_fields
    • BigEarthNetSchema
      • BigEarthNetSchema.opts
      • BigEarthNetSchema.fields
      • BigEarthNetSchema.load_fields
      • BigEarthNetSchema.dump_fields
    • SpaceNet6DatasetSchema
      • SpaceNet6DatasetSchema.opts
      • SpaceNet6DatasetSchema.fields
      • SpaceNet6DatasetSchema.load_fields
      • SpaceNet6DatasetSchema.dump_fields
    • BreizhCropsSchema
      • BreizhCropsSchema.opts
      • BreizhCropsSchema.fields
      • BreizhCropsSchema.load_fields
      • BreizhCropsSchema.dump_fields
    • CropsDatasetSchema
      • CropsDatasetSchema.opts
      • CropsDatasetSchema.fields
      • CropsDatasetSchema.load_fields
      • CropsDatasetSchema.dump_fields
    • So2SatDatasetSchema
      • So2SatDatasetSchema.opts
      • So2SatDatasetSchema.fields
      • So2SatDatasetSchema.load_fields
      • So2SatDatasetSchema.dump_fields
  • aitlas.datasets.semantic_segmentation module
    • SemanticSegmentationDataset
      • SemanticSegmentationDataset.schema
      • SemanticSegmentationDataset.labels
      • SemanticSegmentationDataset.color_mapping
      • SemanticSegmentationDataset.name
      • SemanticSegmentationDataset.apply_transformations()
      • SemanticSegmentationDataset.load_dataset()
      • SemanticSegmentationDataset.get_labels()
      • SemanticSegmentationDataset.data_distribution_table()
      • SemanticSegmentationDataset.data_distribution_barchart()
      • SemanticSegmentationDataset.show_image()
  • aitlas.datasets.siri_whu module
    • SiriWhuDataset
      • SiriWhuDataset.url
      • SiriWhuDataset.labels
      • SiriWhuDataset.name
  • aitlas.datasets.so2sat module
    • So2SatDataset
      • So2SatDataset.url
      • So2SatDataset.name
      • So2SatDataset.schema
      • So2SatDataset.labels
      • So2SatDataset.get_labels()
      • So2SatDataset.show_image()
      • So2SatDataset.show_samples()
      • So2SatDataset.show_batch()
      • So2SatDataset.data_distribution_table()
      • So2SatDataset.data_distribution_barchart()
  • aitlas.datasets.spacenet6 module
    • polygon_to_mask()
    • process_image()
    • SpaceNet6Dataset
      • SpaceNet6Dataset.schema
      • SpaceNet6Dataset.load_directory()
      • SpaceNet6Dataset.load_other_folds()
      • SpaceNet6Dataset.load_fold()
      • SpaceNet6Dataset.labels()
      • SpaceNet6Dataset.prepare()
  • aitlas.datasets.uc_merced module
    • UcMercedDataset
      • UcMercedDataset.labels
      • UcMercedDataset.name
  • aitlas.datasets.uc_merced_multilabel module
    • UcMercedMultiLabelDataset
      • UcMercedMultiLabelDataset.url
      • UcMercedMultiLabelDataset.labels
      • UcMercedMultiLabelDataset.name
  • aitlas.datasets.urls module
  • aitlas.datasets.whu_rs19 module
    • WHURS19Dataset
      • WHURS19Dataset.url
      • WHURS19Dataset.labels
      • WHURS19Dataset.name

Transforms

• Transforms
  • aitlas.transforms.big_earth_net module
    • ResizeToTensorNormalizeRGB
      • ResizeToTensorNormalizeRGB.configurables
    • ToTensorResizeRandomCropFlipHV
    • ToTensorResizeCenterCrop
    • ToTensorResize
    • NormalizeAllBands
      • NormalizeAllBands.configurables
    • ToTensorAllBands
  • aitlas.transforms.breizhcrops module
    • SelectBands
      • SelectBands.configurables
  • aitlas.transforms.classification module
    • ResizeRandomCropFlipHVToTensor
    • ResizeCenterCropFlipHVToTensor
    • ResizeCenterCropToTensor
    • Resize1ToTensor
    • GrayToRGB
    • ConvertToRGBResizeCenterCropToTensor
    • RandomFlipHVToTensor
    • ComplexTransform
  • aitlas.transforms.joint_transforms module
    • FlipHVRandomRotate
    • FlipHVToTensorV2
    • ResizeToTensorV2
    • Resize
  • aitlas.transforms.object_detection module
  • aitlas.transforms.segmentation module
    • MinMaxNormTranspose
    • Transpose
    • MinMaxNorm
    • Pad
    • ColorTransformations
    • ResizeToTensor
    • ResizePerChannelToTensor
  • aitlas.transforms.spacenet6 module
    • saturation()
    • brightness()
    • contrast()
    • SpaceNet6Transforms

Models

• Models
  • aitlas.models.alexnet module
    • AlexNet
      • AlexNet.name
      • AlexNet.forward()
      • AlexNet.extract_features()
      • AlexNet.freeze()
      • AlexNet.training
    • AlexNetMultiLabel
      • AlexNetMultiLabel.name
      • AlexNetMultiLabel.forward()
      • AlexNetMultiLabel.extract_features()
      • AlexNetMultiLabel.training
      • AlexNetMultiLabel.freeze()
  • aitlas.models.cnn_rnn module
    • EncoderCNN
      • EncoderCNN.forward()
      • EncoderCNN.training
    • DecoderRNN
      • DecoderRNN.forward()
      • DecoderRNN.training
    • CNNRNN
      • CNNRNN.schema
      • CNNRNN.forward()
      • CNNRNN.training
  • aitlas.models.convnext module
    • ConvNeXtTiny
      • ConvNeXtTiny.name
      • ConvNeXtTiny.forward()
      • ConvNeXtTiny.freeze()
      • ConvNeXtTiny.extract_features()
      • ConvNeXtTiny.training
    • ConvNeXtTinyMultiLabel
      • ConvNeXtTinyMultiLabel.name
      • ConvNeXtTinyMultiLabel.forward()
      • ConvNeXtTinyMultiLabel.extract_features()
      • ConvNeXtTinyMultiLabel.training
      • ConvNeXtTinyMultiLabel.freeze()
  • aitlas.models.deeplabv3 module
    • DeepLabV3
      • DeepLabV3.forward()
      • DeepLabV3.training
  • aitlas.models.deeplabv3plus module
    • DeepLabV3Plus
      • DeepLabV3Plus.forward()
      • DeepLabV3Plus.training
  • aitlas.models.densenet module
    • DenseNet161
      • DenseNet161.name
      • DenseNet161.forward()
      • DenseNet161.extract_features()
      • DenseNet161.freeze()
      • DenseNet161.training
    • DenseNet161MultiLabel
      • DenseNet161MultiLabel.name
      • DenseNet161MultiLabel.training
      • DenseNet161MultiLabel.forward()
      • DenseNet161MultiLabel.extract_features()
      • DenseNet161MultiLabel.freeze()
  • aitlas.models.efficientnet module
    • EfficientNetB0
      • EfficientNetB0.name
      • EfficientNetB0.forward()
      • EfficientNetB0.freeze()
      • EfficientNetB0.extract_features()
      • EfficientNetB0.training
    • EfficientNetB0MultiLabel
      • EfficientNetB0MultiLabel.name
      • EfficientNetB0MultiLabel.forward()
      • EfficientNetB0MultiLabel.extract_features()
      • EfficientNetB0MultiLabel.freeze()
      • EfficientNetB0MultiLabel.training
    • EfficientNetB4
      • EfficientNetB4.name
      • EfficientNetB4.forward()
      • EfficientNetB4.freeze()
      • EfficientNetB4.extract_features()
      • EfficientNetB4.training
    • EfficientNetB4MultiLabel
      • EfficientNetB4MultiLabel.name
      • EfficientNetB4MultiLabel.forward()
      • EfficientNetB4MultiLabel.extract_features()
      • EfficientNetB4MultiLabel.freeze()
      • EfficientNetB4MultiLabel.training
    • EfficientNetB7
      • EfficientNetB7.name
      • EfficientNetB7.forward()
      • EfficientNetB7.freeze()
      • EfficientNetB7.extract_features()
      • EfficientNetB7.training
    • EfficientNetB7MultiLabel
      • EfficientNetB7MultiLabel.name
      • EfficientNetB7MultiLabel.training
      • EfficientNetB7MultiLabel.forward()
      • EfficientNetB7MultiLabel.extract_features()
      • EfficientNetB7MultiLabel.freeze()
  • aitlas.models.efficientnet_v2 module
    • EfficientNetV2
      • EfficientNetV2.name
      • EfficientNetV2.forward()
      • EfficientNetV2.training
    • EfficientNetV2MultiLabel
      • EfficientNetV2MultiLabel.name
      • EfficientNetV2MultiLabel.training
      • EfficientNetV2MultiLabel.forward()
  • aitlas.models.fasterrcnn module
    • FasterRCNN
      • FasterRCNN.forward()
      • FasterRCNN.training
  • aitlas.models.fcn module
    • FCN
      • FCN.forward()
      • FCN.training
  • aitlas.models.hrnet module
    • HRNetModule
      • HRNetModule.forward()
      • HRNetModule.training
    • HRNetSegHead
      • HRNetSegHead.forward()
      • HRNetSegHead.training
    • HRNet
      • HRNet.forward()
      • HRNet.training
  • aitlas.models.inceptiontime module
    • InceptionTime
      • InceptionTime.schema
      • InceptionTime.forward()
      • InceptionTime.load_optimizer()
      • InceptionTime.training
    • InceptionModule
      • InceptionModule.forward()
      • InceptionModule.training
  • aitlas.models.lstm module
    • LSTM
      • LSTM.schema
      • LSTM.logits()
      • LSTM.forward()
      • LSTM.load_optimizer()
      • LSTM.training
  • aitlas.models.mlp_mixer module
    • MLPMixer
      • MLPMixer.name
      • MLPMixer.forward()
      • MLPMixer.training
    • MLPMixerMultilabel
      • MLPMixerMultilabel.name
      • MLPMixerMultilabel.forward()
      • MLPMixerMultilabel.training
  • aitlas.models.msresnet module
    • conv3x3()
    • conv5x5()
    • conv7x7()
    • BasicBlock3x3
      • BasicBlock3x3.expansion
      • BasicBlock3x3.forward()
      • BasicBlock3x3.training
    • BasicBlock5x5
      • BasicBlock5x5.expansion
      • BasicBlock5x5.forward()
      • BasicBlock5x5.training
    • BasicBlock7x7
      • BasicBlock7x7.expansion
      • BasicBlock7x7.forward()
      • BasicBlock7x7.training
    • MSResNet
      • MSResNet.schema
      • MSResNet.forward()
      • MSResNet.load_optimizer()
      • MSResNet.training
  • aitlas.models.omniscalecnn module
    • SampaddingConv1D_BN
      • SampaddingConv1D_BN.forward()
      • SampaddingConv1D_BN.training
    • build_layer_with_layer_parameter
      • build_layer_with_layer_parameter.forward()
      • build_layer_with_layer_parameter.training
    • OmniScaleCNN
      • OmniScaleCNN.schema
      • OmniScaleCNN.forward()
      • OmniScaleCNN.load_optimizer()
      • OmniScaleCNN.training
    • get_Prime_number_in_a_range()
    • get_out_channel_number()
    • generate_layer_parameter_list()
  • aitlas.models.resnet module
    • ResNet50
      • ResNet50.name
      • ResNet50.forward()
      • ResNet50.freeze()
      • ResNet50.extract_features()
      • ResNet50.training
    • ResNet152
      • ResNet152.name
      • ResNet152.forward()
      • ResNet152.extract_features()
      • ResNet152.freeze()
      • ResNet152.training
    • ResNet50MultiLabel
      • ResNet50MultiLabel.name
      • ResNet50MultiLabel.forward()
      • ResNet50MultiLabel.extract_features()
      • ResNet50MultiLabel.freeze()
      • ResNet50MultiLabel.training
    • ResNet152MultiLabel
      • ResNet152MultiLabel.name
      • ResNet152MultiLabel.training
      • ResNet152MultiLabel.forward()
      • ResNet152MultiLabel.extract_features()
      • ResNet152MultiLabel.freeze()
  • aitlas.models.schemas module
    • TransformerModelSchema
      • TransformerModelSchema.opts
      • TransformerModelSchema.fields
      • TransformerModelSchema.load_fields
      • TransformerModelSchema.dump_fields
    • InceptionTimeSchema
      • InceptionTimeSchema.opts
      • InceptionTimeSchema.fields
      • InceptionTimeSchema.load_fields
      • InceptionTimeSchema.dump_fields
    • LSTMSchema
      • LSTMSchema.opts
      • LSTMSchema.fields
      • LSTMSchema.load_fields
      • LSTMSchema.dump_fields
    • MSResNetSchema
      • MSResNetSchema.opts
      • MSResNetSchema.fields
      • MSResNetSchema.load_fields
      • MSResNetSchema.dump_fields
    • TempCNNSchema
      • TempCNNSchema.opts
      • TempCNNSchema.fields
      • TempCNNSchema.load_fields
      • TempCNNSchema.dump_fields
    • StarRNNSchema
      • StarRNNSchema.opts
      • StarRNNSchema.fields
      • StarRNNSchema.load_fields
      • StarRNNSchema.dump_fields
    • OmniScaleCNNSchema
      • OmniScaleCNNSchema.opts
      • OmniScaleCNNSchema.fields
      • OmniScaleCNNSchema.load_fields
      • OmniScaleCNNSchema.dump_fields
    • UnsupervisedDeepMulticlassClassifierSchema
      • UnsupervisedDeepMulticlassClassifierSchema.opts
      • UnsupervisedDeepMulticlassClassifierSchema.fields
      • UnsupervisedDeepMulticlassClassifierSchema.load_fields
      • UnsupervisedDeepMulticlassClassifierSchema.dump_fields
    • UNetEfficientNetModelSchema
      • UNetEfficientNetModelSchema.opts
      • UNetEfficientNetModelSchema.fields
      • UNetEfficientNetModelSchema.load_fields
      • UNetEfficientNetModelSchema.dump_fields
    • CNNRNNModelSchema
      • CNNRNNModelSchema.opts
      • CNNRNNModelSchema.fields
      • CNNRNNModelSchema.load_fields
      • CNNRNNModelSchema.dump_fields
  • aitlas.models.shallow module
    • ShallowCNNNet
      • ShallowCNNNet.forward()
      • ShallowCNNNet.training
    • ShallowCNNNetMultilabel
      • ShallowCNNNetMultilabel.forward()
      • ShallowCNNNetMultilabel.training
  • aitlas.models.starrnn module
    • StarRNN
      • StarRNN.schema
      • StarRNN.forward()
      • StarRNN.load_optimizer()
      • StarRNN.training
    • StarCell
      • StarCell.reset_parameters()
      • StarCell.forward()
      • StarCell.training
    • StarLayer
      • StarLayer.forward()
      • StarLayer.training
  • aitlas.models.swin_transformer module
    • SwinTransformer
      • SwinTransformer.name
      • SwinTransformer.freeze()
      • SwinTransformer.forward()
      • SwinTransformer.training
    • SwinTransformerMultilabel
      • SwinTransformerMultilabel.name
      • SwinTransformerMultilabel.freeze()
      • SwinTransformerMultilabel.forward()
      • SwinTransformerMultilabel.training
  • aitlas.models.tempcnn module
    • TempCNN
      • TempCNN.schema
      • TempCNN.forward()
      • TempCNN.load_optimizer()
      • TempCNN.training
    • Conv1D_BatchNorm_Relu_Dropout
      • Conv1D_BatchNorm_Relu_Dropout.forward()
      • Conv1D_BatchNorm_Relu_Dropout.training
    • FC_BatchNorm_Relu_Dropout
      • FC_BatchNorm_Relu_Dropout.forward()
      • FC_BatchNorm_Relu_Dropout.training
    • Flatten
      • Flatten.forward()
      • Flatten.training
  • aitlas.models.transformer module
    • TransformerModel
      • TransformerModel.schema
      • TransformerModel.forward()
      • TransformerModel.load_optimizer()
      • TransformerModel.training
    • Flatten
      • Flatten.forward()
      • Flatten.training
  • aitlas.models.unet module
    • Unet
      • Unet.forward()
      • Unet.training
  • aitlas.models.unet_efficientnet module
    • post_process()
    • post_process_single()
    • evaluation()
    • FocalLoss2d
      • FocalLoss2d.forward()
      • FocalLoss2d.training
    • DiceLoss
      • DiceLoss.forward()
      • DiceLoss.training
    • GenEfficientNet
      • GenEfficientNet.training
    • UNetEfficientNet
      • UNetEfficientNet.schema
      • UNetEfficientNet.forward()
      • UNetEfficientNet.load_optimizer()
      • UNetEfficientNet.load_lr_scheduler()
      • UNetEfficientNet.train_and_evaluate_model()
      • UNetEfficientNet.evaluate()
      • UNetEfficientNet.load_model()
      • UNetEfficientNet.training
  • aitlas.models.unsupervised module
    • UnsupervisedDeepMulticlassClassifier
      • UnsupervisedDeepMulticlassClassifier.schema
      • UnsupervisedDeepMulticlassClassifier.train_epoch()
      • UnsupervisedDeepMulticlassClassifier.forward()
      • UnsupervisedDeepMulticlassClassifier.training
    • compute_features()
    • VGG
      • VGG.forward()
      • VGG.training
    • make_layers()
    • vgg16()
    • UnifLabelSampler
      • UnifLabelSampler.generate_indexes_epoch()
  • aitlas.models.vgg module
    • VGG16
      • VGG16.name
      • VGG16.forward()
      • VGG16.freeze()
      • VGG16.extract_features()
      • VGG16.training
    • VGG19
      • VGG19.name
      • VGG19.forward()
      • VGG19.extract_features()
      • VGG19.freeze()
      • VGG19.training
    • VGG16MultiLabel
      • VGG16MultiLabel.name
      • VGG16MultiLabel.forward()
      • VGG16MultiLabel.extract_features()
      • VGG16MultiLabel.freeze()
      • VGG16MultiLabel.training
    • VGG19MultiLabel
      • VGG19MultiLabel.name
      • VGG19MultiLabel.training
      • VGG19MultiLabel.forward()
      • VGG19MultiLabel.freeze()
      • VGG19MultiLabel.extract_features()
  • aitlas.models.vision_transformer module
    • VisionTransformer
      • VisionTransformer.name
      • VisionTransformer.freeze()
      • VisionTransformer.forward()
      • VisionTransformer.training
    • VisionTransformerMultilabel
      • VisionTransformerMultilabel.name
      • VisionTransformerMultilabel.freeze()
      • VisionTransformerMultilabel.training
      • VisionTransformerMultilabel.forward()

Tasks

• Tasks
  • aitlas.tasks.evaluate module
    • EvaluateTask
      • EvaluateTask.schema
      • EvaluateTask.run()
  • aitlas.tasks.extract_features module
    • ExtractFeaturesTask
      • ExtractFeaturesTask.schema
      • ExtractFeaturesTask.run()
  • aitlas.tasks.predict module
    • ImageFolderDataset
    • PredictTask
      • PredictTask.schema
      • PredictTask.run()
      • PredictTask.export_predictions_to_csv()
    • PredictSegmentationTask
      • PredictSegmentationTask.schema
      • PredictSegmentationTask.run()
    • PredictEOPatchTask
      • PredictEOPatchTask.schema
      • PredictEOPatchTask.run()
      • PredictEOPatchTask.export_predictions_to_csv()
  • aitlas.tasks.prepare module
    • PrepareTask
      • PrepareTask.schema
      • PrepareTask.run()
  • aitlas.tasks.schemas module
    • BaseTaskShema
      • BaseTaskShema.opts
    • SplitSetObjectSchema
      • SplitSetObjectSchema.opts
    • SplitObjectSchema
      • SplitObjectSchema.opts
    • SplitTaskSchema
      • SplitTaskSchema.opts
      • SplitTaskSchema.fields
      • SplitTaskSchema.load_fields
      • SplitTaskSchema.dump_fields
    • TrainTaskSchema
      • TrainTaskSchema.opts
      • TrainTaskSchema.fields
      • TrainTaskSchema.load_fields
      • TrainTaskSchema.dump_fields
    • TrainAndEvaluateTaskSchema
      • TrainAndEvaluateTaskSchema.opts
      • TrainAndEvaluateTaskSchema.fields
      • TrainAndEvaluateTaskSchema.load_fields
      • TrainAndEvaluateTaskSchema.dump_fields
    • ParameterSchema
      • ParameterSchema.opts
    • OptimizeTaskSchema
      • OptimizeTaskSchema.opts
      • OptimizeTaskSchema.fields
      • OptimizeTaskSchema.load_fields
      • OptimizeTaskSchema.dump_fields
    • EvaluateTaskSchema
      • EvaluateTaskSchema.opts
      • EvaluateTaskSchema.fields
      • EvaluateTaskSchema.load_fields
      • EvaluateTaskSchema.dump_fields
    • PredictTaskSchema
      • PredictTaskSchema.opts
      • PredictTaskSchema.fields
      • PredictTaskSchema.load_fields
      • PredictTaskSchema.dump_fields
    • PrepareTaskSchema
      • PrepareTaskSchema.opts
      • PrepareTaskSchema.fields
      • PrepareTaskSchema.load_fields
      • PrepareTaskSchema.dump_fields
    • ExtractFeaturesTaskSchema
      • ExtractFeaturesTaskSchema.opts
      • ExtractFeaturesTaskSchema.fields
      • ExtractFeaturesTaskSchema.load_fields
      • ExtractFeaturesTaskSchema.dump_fields
    • VisualizeSplitSetObjectSchema
      • VisualizeSplitSetObjectSchema.opts
    • VisualizeSplitObjectSchema
      • VisualizeSplitObjectSchema.opts
    • VisualizeTaskSchema
      • VisualizeTaskSchema.opts
      • VisualizeTaskSchema.fields
      • VisualizeTaskSchema.load_fields
      • VisualizeTaskSchema.dump_fields
  • aitlas.tasks.split module
    • BaseSplitTask
      • BaseSplitTask.schema
      • BaseSplitTask.is_multilabel
      • BaseSplitTask.extensions
      • BaseSplitTask.run()
      • BaseSplitTask.has_val()
      • BaseSplitTask.is_split_valid()
      • BaseSplitTask.split()
      • BaseSplitTask.save_split()
      • BaseSplitTask.load_images()
      • BaseSplitTask.make_splits()
      • BaseSplitTask.perform_split()
    • RandomSplitTask
      • RandomSplitTask.perform_split()
    • StratifiedSplitTask
      • StratifiedSplitTask.perform_split()
  • aitlas.tasks.train module
    • TrainTask
      • TrainTask.schema
      • TrainTask.run()
    • TrainAndEvaluateTask
      • TrainAndEvaluateTask.schema
      • TrainAndEvaluateTask.run()
    • generate_parameters_for_range()
    • generate_parameters()
    • OptimizeTask
      • OptimizeTask.schema
      • OptimizeTask.run()
  • aitlas.tasks.unsupervised_pre_training module
  • aitlas.tasks.visualize module
    • VisualizeTask
      • VisualizeTask.schema
      • VisualizeTask.get_distribution_for_split()
      • VisualizeTask.get_distribution()
      • VisualizeTask.run()

Metrics

• Metrics
  • aitlas.metrics.classification module
    • AccuracyScore
      • AccuracyScore.name
      • AccuracyScore.key
      • AccuracyScore.calculate()
    • AveragedScore
      • AveragedScore.calculate()
    • PrecisionScore
      • PrecisionScore.name
      • PrecisionScore.key
    • RecallScore
      • RecallScore.name
      • RecallScore.key
    • F1Score
      • F1Score.name
      • F1Score.key
  • aitlas.metrics.segmentation module
    • F1ScoreSample
      • F1ScoreSample.name
      • F1ScoreSample.key
      • F1ScoreSample.calculate()
    • IoU
      • IoU.name
      • IoU.key
      • IoU.calculate()
    • Accuracy
      • Accuracy.name
      • Accuracy.key
      • Accuracy.calculate()
    • DiceCoefficient
      • DiceCoefficient.name
      • DiceCoefficient.key
      • DiceCoefficient.calculate()
    • FocalLoss
      • FocalLoss.name
      • FocalLoss.key
      • FocalLoss.calculate()
    • CompositeMetric
      • CompositeMetric.name
      • CompositeMetric.key
      • CompositeMetric.calculate()

Visualizations

• Visualizations
  • aitlas.visualizations.classification module
    • plot_confusion_matrix()
    • plot_multilabel_confusion_matrix()
    • plot_multiclass_confusion_matrix()
    • PrecisionRecallCurve
      • PrecisionRecallCurve.plot()
    • ImageLabelsVisualization
      • ImageLabelsVisualization.plot()
      • ImageLabelsVisualization.plot_prediction()
    • display_image_labels()
    • precision_recall_curve()
  • aitlas.visualizations.eopatch module
    • display_eopatch_predictions()
  • aitlas.visualizations.grad_cam module
    • ActivationsAndGradients
      • ActivationsAndGradients.save_activation()
      • ActivationsAndGradients.save_gradient()
      • ActivationsAndGradients.release()
    • get_2d_projection()
    • scale_cam_image()
    • show_cam_on_image()
    • ClassifierOutputTarget
    • reshape_transform()
    • BaseCAM
      • BaseCAM.get_cam_weights()
      • BaseCAM.get_cam_image()
      • BaseCAM.forward()
      • BaseCAM.get_target_width_height()
      • BaseCAM.compute_cam_per_layer()
      • BaseCAM.aggregate_multi_layers()
      • BaseCAM.forward_augmentation_smoothing()
    • GradCAM
      • GradCAM.get_cam_weights()
  • aitlas.visualizations.segmentation module
    • ImageMaskPredictionVisualization
      • ImageMaskPredictionVisualization.plot()
      • ImageMaskPredictionVisualization.plot_segmenation()
    • display_image_segmentation()
    • save_predicted_masks()

Utils

• Utils
  • aitlas.utils.segmentation_losses module
    • DiceLoss
      • DiceLoss.forward()
      • DiceLoss.training
    • FocalLoss
      • FocalLoss.ALPHA
      • FocalLoss.GAMMA
      • FocalLoss.forward()
      • FocalLoss.training
  • aitlas.utils.utils module
    • get_class()
    • current_ts()
    • pil_loader()
    • tiff_loader()
    • image_loader()
    • image_invert()
    • stringify()
    • parse_img_id()
    • split_images()
    • load_voc_format_dataset()
    • has_file_allowed_extension()
    • load_folder_per_class_dataset()
    • load_aitlas_format_dataset()
    • submit_inria_results()
    • save_best_model()
    • collate_fn()

Clutering

• Clustering
  • aitlas.clustering.kmeans module
    • Kmeans
      • Kmeans.cluster()
  • aitlas.clustering.pic module
    • PIC
      • PIC.cluster()
  • aitlas.clustering.utils module
    • preprocess_features()
    • make_graph()
    • ReassignedDataset
      • ReassignedDataset.make_dataset()
    • cluster_assign()
    • run_kmeans()
    • arrange_clustering()
    • make_adjacencyW()
    • run_pic()
    • find_maxima_cluster()

Run

• Run
  • aitlas.run module
    • run()
    • main()