def test_IOI_NIFTI_predictionhandling(self):
interface = NIFTI_interface(pattern="nifti")
sample_list = interface.initialize(self.tmp_data.name)
sample = MIScnn_sample.Sample("pred.nifti", np.asarray([0]), interface.channels, interface.classes)
sample.add_prediction(self.seg);
interface.save_prediction(sample, self.tmp_data.name)
pred = interface.load_prediction("pred.nifti", self.tmp_data.name)
self.assertTrue(np.array_equal(pred.reshape(self.seg.shape), self.seg))
def test_IOI_NIFTI_predictionhandling(self):
interface = NIFTI_interface(pattern="nifti")
sample_list = interface.initialize(self.tmp_data.name)
interface.save_prediction(self.seg, "pred.nifti", self.tmp_data.name)
pred = interface.load_prediction("pred.nifti", self.tmp_data.name)
self.assertTrue(np.array_equal(pred, self.seg))