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

Instance-level Image Retrieval using Reranking Transformers

Built a deep neural network (DNN) that functions as an end-to-end machine translation pipeline

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2020

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

Object recognition using Azure Custom Vision AI and Azure Functions

CVPR '21: In the light of feature distributions: Moment matching for Neural Style Transfer

YOLOv7 - Framework Beyond Detection

The Self-Supervised Learner can be used to train a classifier with fewer labeled examples needed using self-supervised learning.

A minimalist implementation of score-based diffusion model

Spatially-Adaptive Pixelwise Networks for Fast Image Translation, CVPR 2021

FLSim a flexible, standalone library written in PyTorch that simulates FL settings with a minimal, easy-to-use API

Implements Gradient Centralization and allows it to use as a Python package in TensorFlow

Pytorch implementation of the paper Improving Text-to-Image Synthesis Using Contrastive Learning

This repository provides data for the VAW dataset as described in the CVPR 2021 paper titled "Learning to Predict Visual Attributes in the Wild"

A collection of awesome resources image-to-image translation.

FS-Mol: A Few-Shot Learning Dataset of Molecules

Benchmark spaces - Benchmarks of how well different two dimensional spaces work for clustering algorithms

Code for the prototype tool in our paper "CoProtector: Protect Open-Source Code against Unauthorized Training Usage with Data Poisoning".

Release of the ConditionalQA dataset

Learning Continuous Signed Distance Functions for Shape Representation