UNet-Annotated A fully annotated unet project Environment python >= 3.6 pythorch 1.12 matplotlib numpy Pillow torch torchvision tqdm wandb