def test_select_annotations(config, image):
windows = preprocess.compute_windows(image, config["patch_size"],
config["patch_overlap"])
image_annotations = pd.read_csv("tests/data/OSBS_029.csv")
selected_annotations = preprocess.select_annotations(image_annotations,
windows,
index=7)
# Returns a 5 column matrix
assert selected_annotations.shape[0] == 17
# image name should be name of image plus the index .tif
assert selected_annotations.image_path.unique()[0] == "OSBS_029_7.png"
def test_select_annotations_tile(config, image):
config["patch_size"] = 50
windows = preprocess.compute_windows(image, config["patch_size"],
config["patch_overlap"])
image_annotations = pd.read_csv("tests/data/OSBS_029.csv")
selected_annotations = preprocess.select_annotations(image_annotations,
windows,
index=10)
# The largest box cannot be off the edge of the window
assert selected_annotations.xmin.min() >= 0
assert selected_annotations.ymin.min() >= 0
assert selected_annotations.xmax.max() <= config["patch_size"]
assert selected_annotations.ymax.max() <= config["patch_size"]
