artus package

Subpackages

• artus.evaluate_model package
  • Submodules
  • artus.evaluate_model.coco_stats module
    • COCOStats
      • COCOStats.coco_path
      • COCOStats.min_nb_occurrences
      • COCOStats.export_stats()
      • COCOStats.get_class_stats()
      • COCOStats.get_nb_images()
      • COCOStats.process_coco()
    • rm_min_classes()
    • rm_tiles_without_annot()
  • artus.evaluate_model.evaluate module
    • evaluate_model()
  • artus.evaluate_model.write_eval_results module
    • ModelsMetricsFormat
      • ModelsMetricsFormat.session
      • ModelsMetricsFormat.eval_results
      • ModelsMetricsFormat.export_dir
      • ModelsMetricsFormat.csv_name
      • ModelsMetricsFormat.check_csv_exists()
      • ModelsMetricsFormat.to_pandas()
      • ModelsMetricsFormat.write_to_csv()
    • ModelsMetricsPlots
      • ModelsMetricsPlots.csv_metrics_path
      • ModelsMetricsPlots.export_dir
      • ModelsMetricsPlots.plot_name
      • ModelsMetricsPlots.title
      • ModelsMetricsPlots.export_plots()
      • ModelsMetricsPlots.plot_metrics()
      • ModelsMetricsPlots.process_csv()
  • Module contents
• artus.inference package
  • Submodules
  • artus.inference.config module
    • add_config()
    • check_logs()
  • artus.inference.deploy_unlabeled_dataset module
    • create_or_load_dataset()
  • artus.inference.predict module
    • add_predictions_to_dataset()
    • apply_nms()
    • build_predictor()
    • crop_mask()
    • filter_confidence_labels()
    • filter_small_predictions()
    • predict()
    • predict_on_sample()
  • Module contents
• artus.prepare package
  • Submodules
  • artus.prepare.coco_splitting module
    • COCOSplitter
      • COCOSplitter.coco_path
      • COCOSplitter.export_dir
      • COCOSplitter.coco_train_name
      • COCOSplitter.coco_test_name
      • COCOSplitter.coco_val_name
      • COCOSplitter.min_nb_occurrences
      • COCOSplitter.train_pct
      • COCOSplitter.val_pct
      • COCOSplitter.test_pct
      • COCOSplitter.batch_size
      • COCOSplitter.create_train_test_val_datasets()
      • COCOSplitter.split_coco()
  • artus.prepare.crs_settings module
    • check_crs()
  • artus.prepare.tile module
  • artus.prepare.transform_geojson_to_coco module
    • geojson_to_coco()
  • Module contents
• artus.spatialize package
  • Submodules
  • artus.spatialize.GeoCOCOExporter module
    • GeoCOCOExporter
      • GeoCOCOExporter.coco_path
      • GeoCOCOExporter.epsg_code
      • GeoCOCOExporter.dest_path
      • GeoCOCOExporter.dest_name
      • GeoCOCOExporter.sample_dir
      • GeoCOCOExporter.affine_transform()
      • GeoCOCOExporter.coco2gdf()
      • GeoCOCOExporter.export()
      • GeoCOCOExporter.get_transform()
  • artus.spatialize.GeoFiftyoneExporter module
    • GeoFiftyoneExporter
      • GeoFiftyoneExporter.export_dir
      • GeoFiftyoneExporter.label_type
      • GeoFiftyoneExporter.epsg_code
      • GeoFiftyoneExporter.dest_name
      • GeoFiftyoneExporter.close()
      • GeoFiftyoneExporter.export_sample()
      • GeoFiftyoneExporter.label_cls
      • GeoFiftyoneExporter.log_collection()
      • GeoFiftyoneExporter.requires_image_metadata
      • GeoFiftyoneExporter.setup()
      • GeoFiftyoneExporter.shapely_polygons()
  • artus.spatialize.LocationImporter module
    • import_csv_locations()
  • Module contents
• artus.train package
  • Submodules
  • artus.train.build_trainer module
    • MyTrainer
      • MyTrainer.build_evaluator()
      • MyTrainer.build_hooks()
      • MyTrainer.build_test_loader()
      • MyTrainer.build_train_loader()
  • artus.train.config module
    • add_config()
    • read_config()
  • artus.train.data_augmentation module
    • custom_mapper()
  • artus.train.train module
    • add_logs_path_to_config()
    • train_model()
  • artus.train.validation_hook module
    • LossEvalHook
      • LossEvalHook.after_step()
  • Module contents

Module contents