Giter VIP home page Giter VIP logo

unet-zoo's Introduction

UNET-ZOO

including unet,unet++,attention-unet,r2unet,cenet,segnet ,fcn.

ENVIRONMENT

window10(Ubuntu is OK)+pycharm+python3.6+pytorch1.3.1

HOW TO RUN:

The only thing you should do is enter the dataset.py and correct the path of the datasets. then run ~ example:

python main.py --action train&test --arch UNet --epoch 21 --batch_size 21 

RESULTS

after train and test,3 folders will be created,they are "result","saved_model","saved_predict".

saved_model folder:

After training,the saved model is in this folder.

result folder:

in result folder,there are the logs and the line chart of metrics.such as: image

saved_predict folder:

in this folder,there are the ouput predict of the saved model,such as: image image image

the datasets:

the Cell dataset(dsb2018) link:https://pan.baidu.com/s/1BaVrzYdrSP78CwYaRzZr1w keyword:5l54

the Liver dataset: link:https://pan.baidu.com/s/1FljGCVzu7HPYpwAKvSVN4Q keyword:5l88

the Cell dataset(isbi) link:https://pan.baidu.com/s/1FkfnhU-RnYFZti62-f8AVA keyword:14rz

the Lung dataset: link:https://pan.baidu.com/s/1sLFRmtG2TOTEgUKniJf7AA keyword:qdwo

the Corneal Nerve dataset: link:https://pan.baidu.com/s/1T3-kS_FgYI6DeXv3n1I7bA keyword:ih02

the Eye Vessels(DRIVE dataset) link:https://pan.baidu.com/s/1UkMLmdbM61N8ecgnKlAsPg keyword:f1ek

the Esophagus and Esophagus Cancer dataset from First Affiliated Hospital of Sun Yat-sen University link:https://pan.baidu.com/s/10b5arIQjNpiggwdkgYNHXQ keyword:hivm

unet-zoo's People

Contributors

andy-zhujunwen avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.