在 U-Net 网络中,输入图像的大小可以是任意大小,但输出的维度和对应标签的维度需要与输入图像相同。如果输入图像大小为 (H, W, C),则输出的维度为 (H, W, N),其中 N 是要分割的类别数量。对应的标签的维度也应该为 (H, W, N)。

U-Net 图像分割网络:输入与输出维度解析

原文地址: https://www.cveoy.top/t/topic/oUrF 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录