compute_ASI.py

# Driver program to compute the ASI index of a given Sentinel-2 L2A product

# ==============================================================================

import sys

from ASI import ASI_Index
from detect_sargassum import detect_sargassum

# ==============================================================================
# Program configuration
# See detect_sargassum documentation for further details

# Path to the SAFE directory containing the Sentinel-2 image
dataset_path = "./T16QEJ/S2B_MSIL2A_20190706T160839_N0212_R140_T16QEJ_20190706T201005.SAFE"

# Path to the ASI model to use
model_path = "ASImodelColabv2.h5"

# Output directory
out_dir = "./"

# Mask out pixels using SCL?
apply_mask = True

# List of SCl labels to keep
mask_keep_categs = [6]

# If set to a numeric value, threshold the image using this; the resulting
# image will have only 0 and 1 to represent values below/above the threshold,
# and 2 to represent invalid values
threshold = None

# Save the result as georeferenced GeoTIFF?
save_geotiff = True

# Save the result as a raw numpy array?
save_npy = True

# Save the result as georeferenced OpenJPEG2000? Only when thresholding
save_jp2 = False

# Report actions to screen?
verbose = True

# ==============================================================================

# Initialize ASI index, loading the specified model
index = ASI_Index(model_path=model_path, verbose=verbose)

# Compute index on image
result = detect_sargassum(dataset_path, index, out_dir=out_dir, apply_mask=apply_mask, mask_keep_categs=mask_keep_categs, threshold=threshold, save_npy=save_npy, save_geotiff=save_geotiff, save_jp2=save_jp2, verbose=verbose)

# ==============================================================================