摘要: ## UNet2DModel 整体网络结构 block_out_channels: 参考UNet的思路,收缩阶段图像空间尺寸在变小但特征通道则增加;扩张阶段则相反。 - conv_in: 对输入的像素空间图像进行卷积处理,获得指定通道且与原始图像相同尺寸的第一层特征图 - down_blocks:依 阅读全文
posted @ 2023-08-18 11:25 星辰大海,绿色星球 阅读(1608) 评论(2) 推荐(1) 编辑