This project aims to segment 4 common retinal lesions from Fundus Images.

Last update: Oct 10, 2021

Overview

To train a HEDNet model with conditional GAN to segment Hard Exudates using random seed 765, run python train_gan_ex.py --gan True --seed 765.
To train a HEDNet model with conditional GAN to segment Soft Exudates using random seed 765, run python train_gan_se.py --gan True --seed 765.
To train a HEDNet model with conditional GAN to segment Hemorrhages using random seed 765, run python train_gan_he.py --gan True --seed 765.
To train a HEDNet model with conditional GAN to segment Microaneurysms using random seed 765, run python train_gan_ma.py --gan True --seed 765.
When training HEDNet with cGAN, we apply all 3 preprocessing methods (Denoising + Contrast Enhancement + Brightness Balance).

To evaluate the model on the test set, run python evaluate_model.py --seed 765 --lesion 'MA' --model results/models_ma/model.pth.tar for evaluating a saved model checkpoint on MA under results/ using random seed 765. results/models_ma/model.pth.tar is the directory of the saved model checkpoint.

This project aims to segment 4 common retinal lesions from Fundus Images.

Related tags

Overview

Owner

Husam Nujaim

Pytorch implementation of PTNet for high-resolution and longitudinal infant MRI synthesis

PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)

SynNet - synthetic tree generation using neural networks

The implementation of "Shuffle Transformer: Rethinking Spatial Shuffle for Vision Transformer"

Learning Features with Parameter-Free Layers (ICLR 2022)

Experimenting with computer vision techniques to generate annotated image datasets from gameplay recordings automatically.

Source code for GNN-LSPE (Graph Neural Networks with Learnable Structural and Positional Representations)

[ICCV 2021] Deep Hough Voting for Robust Global Registration

Vision Transformer for 3D medical image registration (Pytorch).

Augmentation for Single-Image-Super-Resolution

CFNet: Cascade and Fused Cost Volume for Robust Stereo Matching（CVPR2021）

Tensorflow python implementation of "Learning High Fidelity Depths of Dressed Humans by Watching Social Media Dance Videos"

An API-first distributed deployment system of deep learning models using timeseries data to analyze and predict systems behaviour

This is the official implementation for the paper "Heterogeneous Multi-player Multi-armed Bandits: Closing the Gap and Generalization" in NeurIPS 2021.

Codes for building and training the neural network model described in Domain-informed neural networks for interaction localization within astroparticle experiments.

Bayesian Neural Networks in PyTorch

Code for WECHSEL: Effective initialization of subword embeddings for cross-lingual transfer of monolingual language models.

Build and run Docker containers leveraging NVIDIA GPUs

Code for Deterministic Neural Networks with Appropriate Inductive Biases Capture Epistemic and Aleatoric Uncertainty

Clockwork Convnets for Video Semantic Segmentation