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

Experiment about Deep Person Re-identification with EfficientNet-v2

InvTorch: memory-efficient models with invertible functions

A resource for learning about ML, DL, PyTorch and TensorFlow. Feedback always appreciated :)

Official Implementation of DDOD (Disentangle your Dense Object Detector), ACM MM2021

Codebase for ECCV18 "The Sound of Pixels"

A wrapper around SageMaker ML Lineage Tracking extending ML Lineage to end-to-end ML lifecycles, including additional capabilities around Feature Store groups, queries, and other relevant artifacts.

Code for How To Create A Fully Automated AI Based Trading System With Python

Code for T-Few from "Few-Shot Parameter-Efficient Fine-Tuning is Better and Cheaper than In-Context Learning"

A pytorch implementation of Reading Wikipedia to Answer Open-Domain Questions.

VGG16 model-based classification project about brain tumor detection.

Gym Threat Defense

Official Tensorflow implementation of U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation (ICLR 2020)

Library for fast text representation and classification.

Official PyTorch Implementation of GAN-Supervised Dense Visual Alignment

MarcoPolo is a clustering-free approach to the exploration of bimodally expressed genes along with group information in single-cell RNA-seq data

Continuous Diffusion Graph Neural Network

Learned model to estimate number of distinct values (NDV) of a population using a small sample.

traiNNer is an open source image and video restoration (super-resolution, denoising, deblurring and others) and image to image translation toolbox based on PyTorch.

Large-Scale Unsupervised Object Discovery

SiT: Self-supervised vIsion Transformer