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

Code for the paper 'A High Performance CRF Model for Clothes Parsing'.

FEDn is an open-source, modular and ML-framework agnostic framework for Federated Machine Learning

Arquitetura e Desenho de Software.

MaskTrackRCNN for video instance segmentation based on mmdetection

Tensorflow Implementation of SMU: SMOOTH ACTIVATION FUNCTION FOR DEEP NETWORKS USING SMOOTHING MAXIMUM TECHNIQUE

A complete end-to-end demonstration in which we collect training data in Unity and use that data to train a deep neural network to predict the pose of a cube. This model is then deployed in a simulated robotic pick-and-place task.

Unofficial implementation of "TTNet: Real-time temporal and spatial video analysis of table tennis" (CVPR 2020)

Some pvbatch (paraview) scripts for postprocessing OpenFOAM data

Step by Step on how to create an vision recognition model using LOBE.ai, export the model and run the model in an Azure Function

Code repository for the paper "Tracking People with 3D Representations"

Code accompanying "Learning What To Do by Simulating the Past", ICLR 2021.

A PyTorch implementation for PyramidNets (Deep Pyramidal Residual Networks)

Code for: Gradient-based Hierarchical Clustering using Continuous Representations of Trees in Hyperbolic Space. Nicholas Monath, Manzil Zaheer, Daniel Silva, Andrew McCallum, Amr Ahmed. KDD 2019.

📖 Deep Attentional Guided Image Filtering

Implementation of Pooling by Sliced-Wasserstein Embedding (NeurIPS 2021)

Official TensorFlow code for the forthcoming paper

Applying PVT to Semantic Segmentation

Classification models 1D Zoo - Keras and TF.Keras

RIM: Reliable Influence-based Active Learning on Graphs.

PyTorch implementation of TSception V2 using DEAP dataset