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

[ICCV 2021] Official Pytorch implementation for Discriminative Region-based Multi-Label Zero-Shot Learning SOTA results on NUS-WIDE and OpenImages

Automate issue discovery for your projects against Lightning nightly and releases.

FaceAnon - Anonymize people in images and videos using yolov5-crowdhuman

This repository contains demos I made with the Transformers library by HuggingFace.

Google Brain - Ventilator Pressure Prediction

Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.

Shuwa Gesture Toolkit is a framework that detects and classifies arbitrary gestures in short videos

DexterRedTool - Dexter's Red Team Tool that creates cronjob/task scheduler to consistently creates users

Official implementation of the paper "AAVAE: Augmentation-AugmentedVariational Autoencoders"

BTC-Generator - BTC Generator With Python

PyTorch implementation of SwAV (Swapping Assignments between Views)

A library for building and serving multi-node distributed faiss indices.

Semi-Supervised Semantic Segmentation with Pixel-Level Contrastive Learning from a Class-wise Memory Bank

Using this codebase as a tool for my own research. Making some modifications to the original repo for my own purposes.

Code of the paper "Performance-Efficiency Trade-offs in Unsupervised Pre-training for Speech Recognition"

Accuracy Aligned. Concise Implementation of Swin Transformer

Nerf pl - NeRF (Neural Radiance Fields) and NeRF in the Wild using pytorch-lightning

Gender Classification Machine Learning Model using Sk-learn in Python with 97%+ accuracy and deployment

Voice Conversion Using Speech-to-Speech Neuro-Style Transfer

FeTaQA: Free-form Table Question Answering