Code Monkey home page Code Monkey logo

ct_denoise_use_dwconvresnet's Introduction

CT_denoise_use_DWConvResNet

This is a CT denoise model. In reality, the projection data will have noise, so it need to design a model to denoise the final reconstruction result.

In our experiment, we add noise into the projection data, and then use FBP algorithm to reconstruction, after that use the DWConvResNet to train by the reconstruction result with noise, compare with the clean reconstruction result.

The DWConvResNet is base ResNet but we remove the pooling layer and change conv to depthwise conv(in tflearn, the depthwise conv is grouped_conv_2d), and add the Channel Shuffle between two depthwise conv. We define a new resnet_dwconv_block like below:

bn->relu->depthwiseconv->ChannelShuffle->bn->relu->depthwiseconv->bn->relu->conv

And the network basic structure is:

   resnet_dwconv_block(outputchannels=64)
   resnet_dwconv_block(outputchannels=128)
   resnet_dwconv_block(outputchannels=256)
   resnet_dwconv_block(outputchannels=128)
   resnet_dwconv_block(outputchannels=64)

The loss is begin with 25549, final loss is 106.2. Here is the loss curve: image

This model can be trained by denosie the gaussian noise, the result like this: The first one is the original picture, second one is add gaussian nosie, the third one is the reconstruction result by using DWConvResNet image

And alse can be trained by denoise the multi-noise(like gaussian noise, poisson nosie and salt&pepper nosie), train loss like this: image Loss at begin is 26390, the final loss is 20.77. Result like below: The first column is orignial picture, second column is add noise(first one is add poisson noise, second one is add gaussian noise, third one is add salt&pepper noise), third column is the denoise picutre. image

ct_denoise_use_dwconvresnet's People

Contributors

ppaulggit avatar

Stargazers

Carpe_diem-Vampire 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.