Code Monkey home page Code Monkey logo

dual-octconv's Introduction

Dual-Octave Convolution for Fast Parallel MR Image Reconstruction (AAAI 2021)

Dependencies

  • Python 3.7
  • Tensorflow 1.14
  • numpy
  • h5py
  • skimage
  • matplotlib
  • tqdm

Install dependencies as follows:

wget https://repo.anaconda.com/archive/Anaconda3-2020.07-Linux-x86_64.sh
bash Anaconda3-2020.07-Linux-x86_64.sh.sh
source ~/.bashrc
conda install python=3.7
conda install tensorflow-gpu==1.14
conda install numpy
conda install scikit-learn
conda install scikit-image
conda install tqdm
conda install opencv

Dataset and Prepartion

All data that we used for our experiments are released at GLOBUS(https://app.globus.org/file-manager?origin_id=15c7de28-a76b-11e9-821c-02b7a92d8e58&origin_path=%2F). Before training, we recommend you to process data into .tfrecords to accelerate the progress. File ./data_preparation/data2tfrecords.py specifies the route of data processing.

How to train and test on Dual-OctConv

Unpack the dataset file to the folder you defined. Then, change the data_dst argument in ./option.py to the place where datasets are located.

Enter in the folder /Dual-OctConv/code

Train

CUDA_VISIBLE_DEVICES=0 python run.py --n_GPU 1 --name OctComplex_B10_a0.125_cpd320_1Un3 --n_blocks 10 --n_feats 64 --lr 1e-3 --alpha 0.125 --data_dst coronal_pd_320 --epoch 50 --mask_name 1Un3_320

Test

CUDA_VISIBLE_DEVICES=0 python tester.py --n_GPU 1 --rsname OctComplex_B10_a0.125_cpd320_1Un3 --n_blocks 10 --n_feats 64 --alpha 0.125 --data_dst coronal_pd_320 --mask_name 1Un3_320 --test_only --save_gt --save_results

Citation

If you find Dual-OctConv useful for your research, please consider citing the following papers:

@inproceedings{feng2021DualOctConv,
  title={Dual-Octave Convolution for Fast Parallel MR Image Reconstruction},
  author={Feng, Chun-Mei and Yang, Zhanyuan and Chen, Geng and Xu, Yong and Shao, Ling},
  booktitle={Proceedings of the 35th AAAI Conference on Artificial Intelligence (AAAI)},
  year={2021}
}

dual-octconv's People

Contributors

chunmeifeng avatar

Watchers

 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.