DL4J: Evaluation class - add 4d (segmentation) support

[AlexDBlack]

Maybe also 4d support for RegressionEvaluation, etc. And ROCMultiClass support.

From gitter:

Alex Black @AlexDBlack 12:06

so the problem here is that the Evaluation class isn't set up to support semantic segmentation, only output of OutputLayer and RnnOutputLayer (i.e., rank2/3 output/labels)
one workaround is reshaping: start with your [n,c,h,w] shape labels/output, permute to [n,h,w,c] then reshape to [n*h*w,c] and call evaluation
all up that looks like INDArray labels2d = labels.permute(0,2,3,1).dup().reshape('c',n*h*w,c) and same for the labels (and yes, the dup there is necessary)
then you can do: Evaluation e = new Evaluation(); e.eval(labels2d, out2d); (with a loop over all minibatches in your dataset)

Aha! Link: https://skymindai.aha.io/features/DL4J-8

[lock]

This thread has been automatically locked since there has not been any recent activity after it was closed. Please open a new issue for related bugs.