颜色分类leetcode SeqNet:血管分割和动脉/静脉分类的联合学习

颜色分类leetcode序列网血管分割和动脉/静脉分类的联合学习视网膜成像是诊断各种疾病的宝贵工具。然而，即使对于有经验的专家来说，阅读视网膜图像也是一项困难且耗时的任务。自动视网膜图像分析的基本步骤是血管分割和动脉/静脉分类，它们提供有关潜在疾病的各种信息。为了提高现有的视网膜图像分析自动化方法的性能，我们提出了两步血管分类。我们采用基于UNet的模型SeqNet从背景中准确地分割血管并对血管类型进行预测。我们的模型按顺序进行分割和分类，这减轻了标签分布偏差的问题并促进了训练。使用时，数据集应放置在./data/ALL，遵循./utils/prepare_dataset.py定义的数据结构。训练命令：python train.py，模型将保存在./trained_model/ ，结果将保存在./output/ 。预测命令：python predict.py -i ./data/test_images/ -o ./output/。