/Genetic-U-Net

The manuscript titled "Genetic U-Net: Automatically Designed Deep Networks for Retinal Vessel Segmentation Using a Genetic Algorithm" has been accepted in IEEE Transactions on Medical Imaging.

Primary LanguagePythonMIT LicenseMIT

Genetic-U-Net

The manuscript titled "Genetic U-Net: Automatically Designed Deep Networks for Retinal Vessel Segmentation Using a Genetic Algorithm" has been accepted in IEEE Transactions on Medical Imaging.

Requirements:torch >= 1.1.0, torchvision >= 0.3.0, deap == 1.3.0, python >= 3.5, etc.

The gene of the best architecture searched on DRIVE is [0, 1, 1, 1, 0, 0, 1, 1, 0, 0, 1, 0, 0, 1, 0, 0, 1, 1, 1, 0, 1, 1, 0, 0, 0, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 1, 0, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 0, 0, 1, 0, 0, 0, 1, 1, 1, 0, 1, 1, 1, 1, 0, 0, 0, 0, 1, 0, 0, 1, 1, 1, 1, 0, 1, 1, 0, 0, 1, 1, 1, 0, 0, 0, 0, 1, 0, 0, 0, 1, 1, 0, 0, 0, 1, 1, 1, 0, 1]