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

Benchmarking Pipeline for Prediction of Protein-Protein Interactions

This is the official Pytorch implementation of the paper "Diverse Motion Stylization for Multiple Style Domains via Spatial-Temporal Graph-Based Generative Model"

Project ArXiv Citation Network

Code for the paper "JANUS: Parallel Tempered Genetic Algorithm Guided by Deep Neural Networks for Inverse Molecular Design"

Python Assignments for the Deep Learning lectures by Andrew NG on coursera with complete submission for grading capability.

Automatically replace ONNX's RandomNormal node with Constant node.

Focal and Global Knowledge Distillation for Detectors

MobileNetV1-V2，MobileNeXt，GhostNet，AdderNet，ShuffleNetV1-V2，Mobile+ViT etc.

Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups

SMPLpix: Neural Avatars from 3D Human Models

Official repository for "Intriguing Properties of Vision Transformers" (2021)

PyTorch Implementation of PortaSpeech: Portable and High-Quality Generative Text-to-Speech

The toolkit to generate auto labeled datasets

A Python-based development platform for automated trading systems - from backtesting to optimisation to livetrading.

The source code of CVPR17 'Generative Face Completion'.

CCP dataset from Clothing Co-Parsing by Joint Image Segmentation and Labeling

Pytorch implementation of "M-LSD: Towards Light-weight and Real-time Line Segment Detection"

Implementation of DocFormer: End-to-End Transformer for Document Understanding, a multi-modal transformer based architecture for the task of Visual Document Understanding (VDU)

Code for EmBERT, a transformer model for embodied, language-guided visual task completion.

HiFi-GAN: High Fidelity Denoising and Dereverberation Based on Speech Deep Features in Adversarial Networks