
图像分割
文章平均质量分 91
图像语义分割、视频分割
lz739337660
这个作者很懒,什么都没留下…
展开
-
TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation (transformer + unet)
TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation 作者单位:JHU、UESTC、Stanford等 代码:https://github.com/Beckschen/TransUNet 论文:https://arxiv.org/pdf/2102.04306.pdf Abstract U-Net缺陷:由于卷积运算的固有局部性,建模远程依赖存在局限性 transformer:已经作为具有先天全局自我注意力原创 2021-03-13 16:40:14 · 2711 阅读 · 0 评论 -
pytorch: DiceLoss MulticlassDiceLoss
pytorch的自定义多类dice_loss 和单类dice_loss: import torch import torch.nn as nn class DiceLoss(nn.Module): def __init__(self): super(DiceLoss, self).__init__() def forward(self, input, target): N = ...原创 2019-04-25 10:50:09 · 1448 阅读 · 3 评论