Skip to content

watchmexiang/UNet-CRF-RNN

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Edge-aware FCN with CRF-RNN layer

change the model in train.py and eval.py

modelFns = { 'unet':Models.VanillaUnet.VanillaUnet, 'segnet':Models.Segnet.Segnet , 'vgg_unet':Models.VGGUnet.VGGUnet , 'vgg_unet2':Models.VGGUnet.VGGUnet2 , 'fcn8':Models.FCN8.FCN8, 'fcn32':Models.FCN32.FCN32, 'crfunet':Models.CRFunet.CRFunet }

data hierarchy ''' Use the Keras data generators to load train and test Image and label are in structure: train/ img/ 0/ gt/ 0/

    test/
        img/
            0/
        gt/
            0/

'''

Usage

  • Train your model

    python train.py --data_path ./datasets/ --checkpoint_path ./checkpoints/
  • Visualize the train loss, dice score, learning rate, output mask, and first layer convolutional kernels per iteration in tensorboard

    tensorboard tensorboard --logdir=./checkpoints
  • Evaluate your model

    python eval.py --data_path ./datasets/ --load_from_checkpoint ./checkpoints/model-0

About

Edge-aware FCN with CRF-RNN layer

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 80.4%
  • C++ 18.4%
  • Makefile 1.2%