GithubHelp home page GithubHelp logo

timonso / fr-unet Goto Github PK

View Code? Open in Web Editor NEW

This project forked from lseventeen/fr-unet

0.0 0.0 0.0 281.78 MB

[JBHI2021] Full-Resolution Network and Dual-ThresholdIteration for Retinal Vessel and CoronaryAngiograph Segmentation

License: MIT License

Python 100.00%

fr-unet's Introduction

PWCPWC

FR-UNet

This repository is the official PyTorch code for the paper 'Full-Resolution Network and Dual-Threshold Iteration for Retinal Vessel and Coronary Angiograph Segmentation'.

Prerequisites

Download our repo:

git clone https://github.com/lseventeen/RF-UNet.git
cd RF-UNet

Install packages from requirements.txt

pip install -r requirements.txt

Datasets processing

Choose a path to create a folder with the dataset name and download datasets DRIVE,CHASEDB1,STARE,CHUAC, and DCA1. Type this in terminal to run the data_process.py file

python data_process.py -dp DATASET_PATH -dn DATASET_NAME

Training

Type this in terminal to run the train.py file

python train.py -dp DATASET_PATH

Test

Type this in terminal to run the test.py file

python test.py -dp DATASET_PATH -wp WEIGHT_FILE_PATH

We have prepared the pre-trained models for both datasets in the folder 'pretrained_weights'. To replicate the results in the paper, directly run the following commands

python test.py -dp DATASET_PATH -wp pretrained_weights/DATASET_NAME

License

This project is licensed under the MIT License - see the LICENSE file for details

fr-unet's People

Contributors

lseventeen 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.