共计1400多张图片,并选用test1中的1102作为测试集,其分割效果可以达到80%左右 另外,model_unet还提供了wide u-net变体和u-net++变体模型,可以使用其进行训练测试。 注:关于u-net的具体信息见论文 ...
代码复现 为了与Swin-Unet在相同数据集和条件下训练,适配输入224*224单通道图片。修改部分参数,Pytorch代码如下: 先对input图像进行Patch Partition,对应于代码里PatchEmbed类的实现,将图片切成patch ...