Skip to content

Munzeltje/unet_medical_image_segmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Simple (pretrained) unet training code with histopathology

Two pretrained architectures were mainly used: DenseNet-161 encoding Unet and Efficientnets

WSI slicer is in /utils

example run:

python train.py --batch_size 24 --model_type dense --n_eval 1000

check the utils/config.py for configs

data loader assumes patches for loading

validation is done using DICE(f1)-score

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages