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

Toolkit for collecting and applying prompts

Official Pytorch Implementation of Unsupervised Image Denoising with Frequency Domain Knowledge

This repository contains the implementation of the paper Contrastive Instance Association for 4D Panoptic Segmentation using Sequences of 3D LiDAR Scans

Computer Vision and Pattern Recognition, NUS CS4243, 2022

Meta Learning for Semi-Supervised Few-Shot Classification

Deep Learning Slide Captcha

FFTNet vocoder implementation

DM-ACME compatible implementation of the Arm26 environment from Mujoco

Videocaptioning.pytorch - A simple implementation of video captioning

Discriminative Region Suppression for Weakly-Supervised Semantic Segmentation

Code for CVPR2019 paper《Unequal Training for Deep Face Recognition with Long Tailed Noisy Data》

Tutorial on scikit-learn and IPython for parallel machine learning

Tensorflow implementation for "Improved Transformer for High-Resolution GANs" (NeurIPS 2021).

Code for our paper Aspect Sentiment Quad Prediction as Paraphrase Generation in EMNLP 2021.

Pytorch implementation of CoCon: A Self-Supervised Approach for Controlled Text Generation

From this paper "SESNet: A Semantically Enhanced Siamese Network for Remote Sensing Change Detection"

RefineMask (CVPR 2021)

Official PyTorch implementation for paper "Efficient Two-Stage Detection of Human–Object Interactions with a Novel Unary–Pairwise Transformer"

Evaluation framework for testing segmentation networks in PyTorch

A simple Tensorflow based library for deep and/or denoising AutoEncoder.