maxpietsch / dstripe Goto Github PK

Hi there, I am trying to use this software for removal of a consistent stripe artefact in all b0 and DWI's .

The code I ran was: docker run --rm --volume /data/bmol239/Analysis/trouble-shooting:/data/bmol239/Analysis/trouble-shooting maxpietsch/dstripe:1.1 dwidestripe /data/bmol239/Analysis/trouble-shooting/dwi.mif /data/bmol239/Analysis/trouble-shooting/mask.mif /data/bmol239/Analysis/trouble-shooting/dstripe_field.mif -device cpu -corrected dwi_ds.mif -debug

The error message :

MRLoader load_np_funs: {'source': <function get_all at 0x7f722fa40d40>, 'target': <function get_all at 0x7f722fa40d40>}
MRLoader load_sample_funs: <function split_by_vol at 0x7f722fa3c200>
loading data, cropped to mask: False
{'mask_source': 'nn/mask.mif', 'source': 'nn/amp.mif'}
Traceback (most recent call last):
File "/opt/dStripe/dstripe/eval_stripes3.py", line 495, in
transforms_val=SampleToTensor4D(), nsamples=0, poverride_dict=poverride_dict)
File "/opt/dStripe/dstripe/trainer.py", line 525, in predict_val
self.__load_validation_data(transforms=transforms_val)
File "/opt/dStripe/dstripe/trainer.py", line 197, in __load_validation_data
self.__load_data('val', transforms)
File "/opt/dStripe/dstripe/trainer.py", line 180, in __load_data
memmap=self.p.dict.get('memmap', False))
File "/opt/dStripe/dstripe/dataloader2.py", line 143, in init
for _im, _md in self.__postproc(imdata, md):
File "/opt/dStripe/dstripe/dataloader2.py", line 184, in __postproc
for imdat, mdat in gen:
File "/opt/dStripe/dstripe/dwitools.py", line 135, in gen
im, md = normalise_fun(im, md)
File "/opt/dStripe/dstripe/dwitools.py", line 94, in normalise_fun
im['source'] *= v
numpy.core._exceptions.UFuncTypeError: Cannot cast ufunc 'multiply' output from dtype('float64') to dtype('int16') with casting rule 'same_kind'
dwidestripe: /opt/env/bin/python3 /opt/dStripe/dstripe/eval_stripes3.py /opt/dStripe/models/dstripe_2019_07_03-31_v2.pth.tar.json nn/amp.mif nn/mask.mif --butterworth_samples_cutoff=0.65625 --outdir=/dwidestripe-tmp-K0VX2U/ --verbose=0 --batch_size=1 --write_field=true --write_corrected=false --slice_native=false --attention --device=cpu

dwidestripe: [ERROR] failed with return code 1