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

Minimal implementation of Denoised Smoothing: A Provable Defense for Pretrained Classifiers in TensorFlow.

Time-stretch audio clips quickly with PyTorch (CUDA supported)! Additional utilities for searching efficient transformations are included.

Out-of-Domain Human Mesh Reconstruction via Dynamic Bilevel Online Adaptation

Codes for "CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series Imputation"

.NET bindings for the Pytorch engine

Video Swin Transformer - PyTorch

Mixed Transformer UNet for Medical Image Segmentation

The official pytorch implementation of our paper "Is Space-Time Attention All You Need for Video Understanding?"

Style-based Point Generator with Adversarial Rendering for Point Cloud Completion (CVPR 2021)

Graph Convolutional Neural Networks with Data-driven Graph Filter (GCNN-DDGF)

Source code for "Taming Visually Guided Sound Generation" (Oral at the BMVC 2021)

Dynamic vae - Dynamic VAE algorithm is used for anomaly detection of battery data

Predict halo masses from simulations via graph neural networks

Tiny Object Detection in Aerial Images.

FADNet++: Real-Time and Accurate Disparity Estimation with Configurable Networks

A PyTorch implementation of "Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networks" (KDD 2019).

Implementation of SegNet: A Deep Convolutional Encoder-Decoder Architecture for Semantic Pixel-Wise Labelling

Drslmarkov - Distributionally Robust Structure Learning for Discrete Pairwise Markov Networks

Train robotic agents to learn pick and place with deep learning for vision-based manipulation in PyBullet.

Cleaned up code for DSTC 10: SIMMC 2.0 track: subtask 2: multimodal coreference resolution