API documentation#
Base#
- Base
- aitlas.base.classification module
- aitlas.base.config module
- 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
RunningScore
MultiClassRunningScore
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
ObjectDetectionRunningScore
- 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
- aitlas.base.segmentation module
- aitlas.base.tasks module
- aitlas.base.transforms module
- aitlas.base.visualizations module
Datasets#
- Datasets
- aitlas.datasets.aid module
- aitlas.datasets.aid_multilabel module
- aitlas.datasets.airs module
- aitlas.datasets.amazon_rainforest module
- 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
- aitlas.datasets.breizhcrops module
DownloadProgressBar
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
- aitlas.datasets.chactun module
- aitlas.datasets.clrs module
- 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
- aitlas.datasets.eopatch_crops module
- aitlas.datasets.eurosat module
- aitlas.datasets.inria module
- aitlas.datasets.landcover_ai module
- aitlas.datasets.massachusetts_buildings module
- aitlas.datasets.massachusetts_roads module
- aitlas.datasets.mlrs_net module
- 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
- aitlas.datasets.object_detection module
- aitlas.datasets.optimal_31 module
- aitlas.datasets.pattern_net module
- aitlas.datasets.planet_uas module
- aitlas.datasets.resisc45 module
- aitlas.datasets.rsd46_whu module
- aitlas.datasets.rsi_cb256 module
- aitlas.datasets.rsscn7 module
- aitlas.datasets.sat6 module
- aitlas.datasets.schemas module
- 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
- aitlas.datasets.so2sat module
- aitlas.datasets.spacenet6 module
- aitlas.datasets.uc_merced module
- aitlas.datasets.uc_merced_multilabel module
- aitlas.datasets.urls module
- aitlas.datasets.whu_rs19 module
Transforms#
Models#
- Models
- aitlas.models.alexnet module
- aitlas.models.cnn_rnn module
- aitlas.models.convnext module
- aitlas.models.deeplabv3 module
- aitlas.models.deeplabv3plus module
- aitlas.models.densenet module
- aitlas.models.efficientnet module
- aitlas.models.efficientnet_v2 module
- aitlas.models.fasterrcnn module
- aitlas.models.fcn module
- aitlas.models.hrnet module
- aitlas.models.inceptiontime module
- aitlas.models.lstm module
- aitlas.models.mlp_mixer module
- aitlas.models.msresnet module
- aitlas.models.omniscalecnn module
- aitlas.models.resnet module
- aitlas.models.schemas module
- aitlas.models.shallow module
- aitlas.models.starrnn module
- aitlas.models.swin_transformer module
- aitlas.models.tempcnn module
- aitlas.models.transformer module
- aitlas.models.unet module
- aitlas.models.unet_efficientnet module
- aitlas.models.unsupervised module
- aitlas.models.vgg module
- aitlas.models.vision_transformer module
Tasks#
- Tasks
- aitlas.tasks.evaluate module
- aitlas.tasks.extract_features module
- aitlas.tasks.predict module
- aitlas.tasks.prepare module
- aitlas.tasks.schemas module
BaseTaskShema
SplitSetObjectSchema
SplitObjectSchema
SplitTaskSchema
TrainTaskSchema
TrainAndEvaluateTaskSchema
ParameterSchema
OptimizeTaskSchema
EvaluateTaskSchema
PredictTaskSchema
PrepareTaskSchema
ExtractFeaturesTaskSchema
VisualizeSplitSetObjectSchema
VisualizeSplitObjectSchema
VisualizeTaskSchema
- aitlas.tasks.split module
- aitlas.tasks.train module
- aitlas.tasks.unsupervised_pre_training module
- aitlas.tasks.visualize module
Metrics#
Visualizations#
- Visualizations
- aitlas.visualizations.classification module
- aitlas.visualizations.eopatch module
- aitlas.visualizations.grad_cam module
- aitlas.visualizations.segmentation module