Documentation for bulk_inference

Path: image_classification/bulk_inference.py

Module Description

Bulk inference for image classification models.

Takes GeoJSON files from input folder, downloads satellite imagery, and classifies each site using a trained model.

Edit the config section at the top before running.

Functions

get_model

Get model with correct output classes

load_model

Load trained model from checkpoint

run_inference_on_image

Run inference on single image

main

Main inference function

main_wrapper

Command line entry point