Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
BUG: Fix error in JSON serialization
Currently, the standard JSON encoder does not serialize int16 and throws a type error. To prevent this, ensure that image and mask diagnostics return type float (for features, this is already the case).
  • Loading branch information
JoostJM committed Oct 15, 2018
1 parent 167888b commit 9d992fe
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions radiomics/generalinfo.py
Expand Up @@ -67,7 +67,7 @@ def addImageElements(self, image, prefix='original'):

self.generalInfo[self.generalInfo_prefix + 'Image-' + prefix + '_Spacing'] = image.GetSpacing()
self.generalInfo[self.generalInfo_prefix + 'Image-' + prefix + '_Size'] = image.GetSize()
im_arr = sitk.GetArrayFromImage(image)
im_arr = sitk.GetArrayFromImage(image).astype('float')
self.generalInfo[self.generalInfo_prefix + 'Image-' + prefix + '_Mean'] = numpy.mean(im_arr)
self.generalInfo[self.generalInfo_prefix + 'Image-' + prefix + '_Minimum'] = numpy.min(im_arr)
self.generalInfo[self.generalInfo_prefix + 'Image-' + prefix + '_Maximum'] = numpy.max(im_arr)
Expand Down Expand Up @@ -132,7 +132,7 @@ def addMaskElements(self, image, mask, label, prefix='original'):
return

im_arr = sitk.GetArrayFromImage(image)
targetvoxels = im_arr[ma_arr]
targetvoxels = im_arr[ma_arr].astype('float')
self.generalInfo[self.generalInfo_prefix + 'Mask-' + prefix + '_Mean'] = numpy.mean(targetvoxels)
self.generalInfo[self.generalInfo_prefix + 'Mask-' + prefix + '_Minimum'] = numpy.min(targetvoxels)
self.generalInfo[self.generalInfo_prefix + 'Mask-' + prefix + '_Maximum'] = numpy.max(targetvoxels)
Expand Down

0 comments on commit 9d992fe

Please sign in to comment.