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

DivNoising is an unsupervised denoising method to generate diverse denoised samples for any noisy input image. This repository contains the code to reproduce the results reported in the paper https://openreview.net/pdf?id=agHLCOBM5jP

Robbing the FED: Directly Obtaining Private Data in Federated Learning with Modified Models

Some code of the implements of Geological Modeling Using 3D Pixel-Adaptive and Deformable Convolutional Neural Network

Code release for Local Light Field Fusion at SIGGRAPH 2019

Code Release for Learning to Adapt to Evolving Domains

Code related to the manuscript "Averting A Crisis In Simulation-Based Inference"

Deep Sea Treasure Environment for Multi-Objective Optimization Research

Deep learning model for EEG artifact removal

Doods2 - API for detecting objects in images and video streams using Tensorflow

Tensorflow implementation for Self-supervised Graph Learning for Recommendation

Model search is a framework that implements AutoML algorithms for model architecture search at scale

[CVPR 21] Vectorization and Rasterization: Self-Supervised Learning for Sketch and Handwriting, IEEE Conf. on Computer Vision and Pattern Recognition (CVPR), 2021.

Weakly-supervised object detection.

A python package for generating, analyzing and visualizing building shadows

Complete system for facial identity system. Include one-shot model, database operation, features visualization, monitoring

ScaleNet: A Shallow Architecture for Scale Estimation

Brain tumor detection using Convolution-Neural Network (CNN)

🎯 A comprehensive gradient-free optimization framework written in Python

Automatically align face images 🙃→🙂. Can also do windowing and warping.

Pacman-AI - AI project designed by UC Berkeley. Designed reflex and minimax agents for the game Pacman.