Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

elevate project to the next level #9

Closed
wants to merge 8 commits into from
Closed

elevate project to the next level #9

wants to merge 8 commits into from

Conversation

bodokaiser
Copy link
Owner

@bodokaiser bodokaiser commented Mar 10, 2017

1. Precision

To increase image precision on MRI images we would need to:

  • read slices from MINC2 as np.float64
  • update network models to use weights, etc. as LongTensors
  • update scripts/patch.py to save images as TIFFs
  • update MNIBITEFolder to load TIFF patches

2. Filtering

To support our enhanced loss function we need to create a binary mask from US thresholding. That mask can also be used to filter too small US images (saves us one calculation) in the training loop.

  • create helper function which creates binary mask
  • filter provided (mr, us) batch in training loop according to binary mask

3. Enhanced Loss

  • use l2 loss
  • find better place for calculating mask
  • support batches
  • use matrix multiplication with mask
  • normalize loss by dividing through pixels

Bugs

  • data loader varnishes after one epoch
  • use same dataset for show images and training

@bodokaiser
Copy link
Owner Author

I just noticed that our ElasticNet implementation is numerically unstable (result yields nan's).

@bodokaiser
Copy link
Owner Author

I found the problem with why data varnishes after one epoch. The reason is that MNIBITENative calls to Normalize() overwrite the images such that with a second normalize (over the original value range) it has lost its precision. I try to find a way that MINC2 loads the corresponding image slice directly over h5py.File each time.

@bodokaiser
Copy link
Owner Author

Rebased with master.

@bodokaiser bodokaiser closed this Mar 14, 2017
@bodokaiser bodokaiser deleted the u-net branch March 14, 2017 08:47
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant