I love this project! I agree support for arbitrary grids would add a lot to the user experience. You might not even need ml for that. For instance, look for th

I love this project! I agree support for arbitrary grids would add a lot to the user experience. You might not even need ml for that. For instance, look for the blob of uninterrupted color that occupies the largest contiguous space in the image and interpret cell order within it by a function of their midpoint and filter for size. You could draw missing outside walls via convex hull generation first.