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

A python library for face detection and features extraction based on mediapipe library

mPose3D, a mmWave-based 3D human pose estimation model.

DFFNet: An IoT-perceptive Dual Feature Fusion Network for General Real-time Semantic Segmentation

Representing Long-Range Context for Graph Neural Networks with Global Attention

code for ICCV 2021 paper 'Generalized Source-free Domain Adaptation'

Companion code for the paper "Meta-Learning the Search Distribution of Black-Box Random Search Based Adversarial Attacks" by Yatsura et al.

Facial Image Inpainting with Semantic Control

This repo is a PyTorch implementation for Paper "Unsupervised Learning for Cuboid Shape Abstraction via Joint Segmentation from Point Clouds"

Deep Image Matting implementation in PyTorch

[NeurIPS 2021] "G-PATE: Scalable Differentially Private Data Generator via Private Aggregation of Teacher Discriminators"

An implementation of "Learning human behaviors from motion capture by adversarial imitation"

Fast RFC3339 compliant Python date-time library

Label-Free Model Evaluation with Semi-Structured Dataset Representations

Learning Features with Parameter-Free Layers (ICLR 2022)

PaddleViT: State-of-the-art Visual Transformer and MLP Models for PaddlePaddle 2.0+

The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relating to PyTorch.

A framework for multi-step probabilistic time-series/demand forecasting models

《Dual-Resolution Correspondence Network》(NeurIPS 2020)

Offical implementation for "Trash or Treasure? An Interactive Dual-Stream Strategy for Single Image Reflection Separation".

B2EA: An Evolutionary Algorithm Assisted by Two Bayesian Optimization Modules for Neural Architecture Search