U-Net 图像分割：输出维度和标签维度解析

使用 U-Net 网络进行图像分割时，图像经过 U-Net 输出的维度与输入图像的维度相同，而对应标签的维度也与输入图像的维度相同，因为需要将每个像素点分为不同的类别。例如，如果输入图像的大小为 256×256×3，则 U-Net 网络输出的大小为 256×256×N，其中 N 表示类别数。而对应标签的大小也是 256×256×N。