test_edges.py

import sys

import numpy as np

import gdal
from mycanny import mycanny

input_image = sys.argv[1]
print(input_image)

input_image = gdal.Open(input_image)

input_image = input_image.GetRasterBand(1).ReadAsArray()


print("Got image")
print(np.nanmax(input_image))
print(np.nanmean(input_image))

edges, thresholds = mycanny(input_image, low_threshold=0.7, high_threshold=0.85, mask=np.isfinite(input_image))

print("Using calculated thresholds of: ")
print(thresholds)
print("Found %d edges" % edges[edges > 0].sum())

edges, thresholds = mycanny(input_image, low_threshold=thresholds[0],
                    high_threshold=thresholds[1], abs_thresh=True, mask=np.isfinite(input_image))

print("Using absolute thresholds of: ")
print(thresholds)
print("Found %d edges" % edges[edges > 0].sum())