Unet-Segmentation-Pytorch-Nest-of-Unets
python图像分割模型_用于图像分割的各种Unet模型的PyTorch实现
最新推荐文章于 2024-09-13 22:29:31 发布
这是一个包含不同Unet模型实现的PyTorch项目,包括UNet、RCNN-UNet、Attention UNet、Attention-RCNN UNet和Nested UNet。提供了模型的训练、评估和可视化功能,支持损失曲线的绘制和特定层的可视化。项目适用于医学图像分割任务,例如在ADNI-LONI数据集上进行海马体分割,达到了较高的Dice分数。
摘要由CSDN通过智能技术生成