search

TooTouch / 3D_CoordConv_Segmentation

Grand Challenge 2017 Multi-Modality Whole Heart Segmentation
13 stars 0 forks source link

Model evaluation #6

Open TooTouch opened 5 years ago

TooTouch commented 5 years ago

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

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

TooTouch commented 5 years ago

추가로 weighted_label_dice_coefficient_loss를 주었음

TooTouch commented 5 years ago

CT가 MR보다 더 심장이 많이 보여져서 그런지 학습속도는 확실히 더 빠르다.

MR

Epoch 19/500 - 14s

CT

Epoch 19/30 - 14s

TooTouch commented 5 years ago

mask label 값이 전처리에서 잘못돼서 학습이 안됐음

TooTouch commented 5 years ago

전처리를 제대로 진행하고 해도 학습이 잘 되지 않는다. 전체 class에 대해서 학습이 되지만 DSC가 10%를 넘는 class가 없다.

그래서 2D로 이미지를 바꿔서 진행하고 있다.

TooTouch commented 5 years ago

2D image가 성능이 잘 나오지 않아서 binary로 background와 아닌것을 기준으로 학습했더니 엄청 잘나왔다.

학습 속도 면에서도 2D U net은 epoch당 160초, 3D U net은 epoch당 30초 정도의 시간이 걸렸다.

한가지 3D U net에서 참고해햐할 부분은 10epoch 정도에서는 크게 학습이 잘 되지 않다가 갑자기 70%로 학습이 확 올랐다.

TooTouch commented 5 years ago

Fine tuning을 해봐야할것같다.