We need to get the rgb visualizations back into the confusion matrix, check the main branch for code under visualize.py