fix to preserve input image shape upon executing `random_walker`