Retinal Image Restoration and Vessel Segmentation using Modified Cycle-CBAM and CBAM-UNet

by   Alnur Alimanov, et al.

Clinical screening with low-quality fundus images is challenging and significantly leads to misdiagnosis. This paper addresses the issue of improving the retinal image quality and vessel segmentation through retinal image restoration. More specifically, a cycle-consistent generative adversarial network (CycleGAN) with a convolution block attention module (CBAM) is used for retinal image restoration. A modified UNet is used for retinal vessel segmentation for the restored retinal images (CBAM-UNet). The proposed model consists of two generators and two discriminators. Generators translate images from one domain to another, i.e., from low to high quality and vice versa. Discriminators classify generated and original images. The retinal vessel segmentation model uses downsampling, bottlenecking, and upsampling layers to generate segmented images. The CBAM has been used to enhance the feature extraction of these models. The proposed method does not require paired image datasets, which are challenging to produce. Instead, it uses unpaired data that consists of low- and high-quality fundus images retrieved from publicly available datasets. The restoration performance of the proposed method was evaluated using full-reference evaluation metrics, e.g., peak signal-to-noise ratio (PSNR) and structural similarity index measure (SSIM). The retinal vessel segmentation performance was compared with the ground-truth fundus images. The proposed method can significantly reduce the degradation effects caused by out-of-focus blurring, color distortion, low, high, and uneven illumination. Experimental results show the effectiveness of the proposed method for retinal image restoration and vessel segmentation.


page 1

page 2

page 3

page 4

page 5

page 6


Retinal Image Restoration using Transformer and Cycle-Consistent Generative Adversarial Network

Medical imaging plays a significant role in detecting and treating vario...

Generation of Structurally Realistic Retinal Fundus Images with Diffusion Models

We introduce a new technique for generating retinal fundus images that h...

Robust and efficient computation of retinal fractal dimension through deep approximation

A retinal trait, or phenotype, summarises a specific aspect of a retinal...

Moire Image Restoration using Multi Level Hyper Vision Net

A moire pattern in the images is resulting from high frequency patterns ...

Efficient Unpaired Image Dehazing with Cyclic Perceptual-Depth Supervision

Image dehazing without paired haze-free images is of immense importance,...

Please sign up or login with your details

Forgot password? Click here to reset