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

TagLab: an image segmentation tool oriented to marine data analysis

This is an open source library implementing hyperbox-based machine learning algorithms

A machine learning malware analysis framework for Android apps.

Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecasting

Meta Self-learning for Multi-Source Domain Adaptation： A Benchmark

Learning Confidence for Out-of-Distribution Detection in Neural Networks

Synthetic Scene Text from 3D Engines

Official repository for the ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology

:boar: :bear: Deep Learning based Python Library for Stock Market Prediction and Modelling

Colab notebook and additional materials for Python-driven analysis of redlining data in Philadelphia

Official code of the paper "Expanding Low-Density Latent Regions for Open-Set Object Detection" (CVPR 2022)

Puzzle-CAM: Improved localization via matching partial and full features.

An experimental technique for efficiently exploring neural architectures.

Auto White-Balance Correction for Mixed-Illuminant Scenes

use tensorflow 2.0 to tell a dog and cat from a specified picture

OBBDetection: an oriented object detection toolbox modified from MMdetection

Realtime segmentation with ENet, the fast and accurate segmentation net.

Baselines for TrajNet++

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

A package for "Procedural Content Generation via Reinforcement Learning" OpenAI Gym interface.