ptrblck:

Would it be possible to pass conf_mat directly to writer.add_image ?
If this method expects a tensor or numpy array, I guess it would be possible or did you get an error?


It does not work, I guess the expected tensor should be a tensor representing an image, while  conf_mat  is a “raw” confusion matrix, e.g.
[[21, 1], 
 [3, 1]]