视网膜血管分割代码(Pytorch实现)

完整博客链接:知乎完整博客链接地址


简介:本文主要分享我在做视网膜血管分割深度学习算法的过程中,整理出来的一套视网膜血管分割代码。简单介绍一下,我的研究方向是计算机视觉(侧重于语义分割),主要做一些医学图像分割算法和偏工程的机器视觉部署方面的工作,具体研究课题是基于深度学习的视网膜血管分割算法研究。在这,把我的视网膜血管分割代码分享给大家,项目地址:

GitHub仓库地址:https://github.com/lee-zq/VesselSeg-Pytorch

该项目是基于pytorch深度学习框架实现的视网膜血管分割代码,包括数据预处理、模型训练、模型测试以及可视化等功能,可以在此基础上进一步研究视网膜血管分割算法。最近我把这套代码进行了重构和简化,在比较晦涩的地方也添加了注释,力求任何一个入门者都能看懂。当然也可能存在bug和表述不清的地方,也希望大家能提issue指明问题,我会尽可能快速debug并更新代码,不胜感激。Besides,Star⭐是我Git push的动力,后续我会搜集经典和SOTA的视网膜血管分割模型进行复现。

可视化分割示例:
视网膜血管分割代码(Pytorch实现)_第1张图片
视网膜血管分割代码(Pytorch实现)_第2张图片
视网膜血管分割代码(Pytorch实现)_第3张图片

你可能感兴趣的:(python,图像处理,深度学习,算法,pytorch)