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

GT China coal model

NLP From Scratch Without Large-Scale Pretraining: A Simple and Efficient Framework

WaveFake: A Data Set to Facilitate Audio DeepFake Detection

EsViT: Efficient self-supervised Vision Transformers

Repo for the Tutorials of Day1-Day3 of the Nordic Probabilistic AI School 2021 (https://probabilistic.ai/)

Code/data of the paper "Hand-Object Contact Prediction via Motion-Based Pseudo-Labeling and Guided Progressive Label Correction" (BMVC2021)

Using Streamlit to host a multi-page tool with model specs and classification metrics, while also accepting user input values for prediction.

A library that can print Python objects in human readable format

Official implementation of "Learning Proposals for Practical Energy-Based Regression", 2021.

phylotorch-bito is a package providing an interface to BITO for phylotorch

Repo for "Physion: Evaluating Physical Prediction from Vision in Humans and Machines" submission to NeurIPS 2021 (Datasets & Benchmarks track)

[NeurIPS-2021] Slow Learning and Fast Inference: Efficient Graph Similarity Computation via Knowledge Distillation

Foreground-Action Consistency Network for Weakly Supervised Temporal Action Localization

StarGAN v2 - Official PyTorch Implementation (CVPR 2020)

Official repo for QHack—the quantum machine learning hackathon

Code for the paper "Learning-Augmented Algorithms for Online Steiner Tree"

LocUNet is a deep learning method to localize a UE based solely on the reported signal strengths from a set of BSs.

[CVPR'2020] DeepDeform: Learning Non-rigid RGB-D Reconstruction with Semi-supervised Data

Deep Halftoning with Reversible Binary Pattern

This project is a loose implementation of paper "Algorithmic Financial Trading with Deep Convolutional Neural Networks: Time Series to Image Conversion Approach"