可以使用自己的数据集,若使用自己的数据集,需要先对label进行voc格式转换,代码位于tools文件夹下voc.py,使用流程为使用train脚本训练网络,使用prediction脚本输出分割结果。图片位于data文件夹下,可以更换为自己的数据集,但需要保持图片为灰度图片,详情见:https://blog.csdn.net/qq_52060635/article/details/134148448?spm=1001.2014.3001.5502
初始任务为医学图像分割,可以用于其他图像处理。
详情见:https://blog.csdn.net/qq_52060635/article/details/134149072?spm=1001.2014.3001.5502
包含滑窗操作,具有层级设计的Swin Transformer滑窗操作包括不重叠的local window,和重叠的cross-window。将注意力计算限制在一个窗口中,一方面能引入CNN卷积操作的局部性,另一方面能节省计算量。
1