LifangHe / retinal-vessel-segmentation-with-rotation-augmentation

The code for paper "DATA AUGMENTATION IS MORE IMPORTANT THAN MODEL ARCHITECTURES FOR RETINAL VESSEL SEGMENTATION " We utilize the retina-unet "https://github.com/orobix/retina-unet" as the basic code, but we change a lot of code and add the rotation augmentation ,which achieves the good result AUC 0.9814.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

retinal-vessel-segmentation-with-rotation-augmentation

The code for "DATA AUGMENTATION IS MORE IMPORTANT THAN MODEL ARCHITECTURES FOR RETINAL VESSEL SEGMENTATION " We utilize the retina-unet "https://github.com/orobix/retina-unet" as the basic code, but we change a lot of code and add the rotation augmentation ,which achieves the good result AUC 0.9814.

To train the model,,

run bash train_1.sh

About

The code for paper "DATA AUGMENTATION IS MORE IMPORTANT THAN MODEL ARCHITECTURES FOR RETINAL VESSEL SEGMENTATION " We utilize the retina-unet "https://github.com/orobix/retina-unet" as the basic code, but we change a lot of code and add the rotation augmentation ,which achieves the good result AUC 0.9814.


Languages

Language:Python 99.8%Language:Shell 0.2%