import torch


def create_simple_image(size, batch=1):
    """
    returns a simple target image with two and a seeds mask
    """
    sizex, sizey = size

    target = torch.zeros(batch, 1, sizex, sizey).long()
    target[:, :, :, sizey//2:] = 1

    seeds = torch.zeros(batch, sizex, sizey).long()
    seeds[:, 3 * sizex//4, 5] = 1
    seeds[:, sizex//4, sizey - 5] = 2

    return seeds, target