urbanlc.model.train_utils

Functions

combine_prediction(preds, coordinates, ...)	Combine predictions from partitioned patches into a complete prediction.
load_checkpoint(checkpoint_path, model[, ...])	Load model checkpoint, including model state, optimizer state, scheduler state, and epoch.
save_checkpoint(save_dir, name, model, ...)	Save model checkpoint, including model state, optimizer state, scheduler state, and epoch.
segment_satelite_image(img[, sub_size, stride])	Partition a satellite image into patches of specified size and stride.
set_seed(seed)	Set random seeds for reproducibility in random, numpy, and PyTorch.