Code Monkey home page Code Monkey logo

3d_coordconv_segmentation's Introduction

Hits

Grand Challenge 2017 Multi-Modality Whole Heart Segmentation

Contribution

Training Run

In code directory

> python main.py --params=ct_train.json

Result

Model Background MLV LABC LVBC RABC RVBC ASA PUA Average DSC
U-net 3D 0.995 0.918 0.929 0.912 0.925 0.923 0.843 0.923 0.909
U-net 3D + CoordConv 0.995 0.919 0.926 0.912 0.933 0.924 0.928 0.897 0.920
  • MLV: the Myocardium of the left ventricle, LABC: the left atrium blood cavity, LVBC: the left ventricle blood cavity, RABC: the right atrium blood cavity, RVBC: the right ventricle blood cavity, ASA: the ascending aorta, PUA: the pulmonary artery
  • Average DSC is average of classes that excluded background
Label 19 U-net 3D U-net 3D + CoordConv

Details

Data Number of train set Number of validation set Patch dim Resize rate Batch size Epochs Number of train patch image Number of validation patch image Metric Loss function Optimizer Learning rate Number of GPU
CT 18 2 96 0.7 2 100 20 100 Dice Similarity Coefficient dice coefficient loss Adam 0.0001 4

Limit

The host server is down, so the test set can no longer be evaluated.

3d_coordconv_segmentation's People

Contributors

forestnoobie avatar heojeahyuk avatar tootouch avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar

3d_coordconv_segmentation's Issues

Negative value in image

zero padding vs min value padding

두 가지의 차이점을 볼 것.
-> 실제 음수값을 갖는 영역이 어떤 영향을 가지고 있는지 확인
-> 한가지 걸리는 점은 min value padding을 할 시 255. 로 rescale이 안된다는 점

Class labels

  • Class가 1개인 경우
    segmentation문제에서는 mask의 pixel값은 0 or 1이기 때문에 sigmoid로 할 수 있었지만

  • Class가 여려개인 경우
    mask의 pixel값은 0~m개가 될 수 있다.(m=1,2,3...) output layer에서 softmax로 7개의 units을 뽑아 낼건데, 그러면 각 픽셀별로 7개의 unit을 가지고 있을거라는 얘기고 그렇다면 마찬가지로 label도 각 pixel별로 7개의 units을 갖게 one-hot-vector로 치환해야하는건지 확인

Model evaluation

학습이 잘 되지 않음. image가 크지않아서 일 수도 있다는 생각이 들긴함.

처음 learning rate를 adam(lr=0.00001)을 두고 돌렸는데 너무 학습이 느린가라고 생각돼서 30 epochs만 확인하고 10배 키운 후 다기 500 epochs 확인. 그러나 거의 학습되지않음

Rescaling pixel values in image

  • Medical image 영역에서는 단순히 255. 로 rescale하는 것은 좋지 않을 수 있다.
    -> 좋지 않다라는 얘기는 들었지만 왜 좋지 않은지가 명확하지 않다. 좋지 않은게 맞는거같은데 제대로된 이해가 필요하다.

  • 그래서 intensity를 rescale할 수 있는 알고리즘 또는 방법이 필요하다.

Image preprocessing

현재

  1. resizing by zero padding
  2. 모든 voxel에 255.로 나누어서 rescale

TODO.

  1. alignment를 위해서 registration을 적용한 후
  2. 심장에 대해서만 crop을 해서 원하는 크기로 resizing
    -> CT의 경우 대부분 심장이지만, MR사진의 경우 영상의 범위가 목~복부까지 넓게 보여진다.

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.