metric script issue
hi,
metric script don't work when predicition and mask files do not have the same number of bands. (assert mask.shape == pred.shape, "Mask shape and prediction shape should be the same.") and this is to restricitve in some usal cases:
the "dataset generation" script always add a n+1 band (called "other" ) and unfortunately, if you decide too train a model excluding this band , mask shape and prediction shape don't feet and metrics dont't work as seen above.
would it be possible that metric script have the same optional param "image_bands" as "train" script ?
Edited by david chandler