Commit 705fc0c8 authored by Guillaume HEUSCH's avatar Guillaume HEUSCH

[extractor] modified type of returned score for DeepPixBiS, now a numpy array...

[extractor] modified type of returned score for DeepPixBiS, now a numpy array to be consistent with MCDeepPixBiS
parent 38a149fa
Pipeline #35592 passed with stage
in 10 minutes and 50 seconds
......@@ -107,5 +107,4 @@ class DeepPixBiSExtractor(Extractor):
raise ValueError('Scoring method {} is not implemented.'.format(self.scoring_method))
# output is a scalar score
return score
return np.reshape(score,(1,-1))
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment