Add class model/saliency visualization; Fix a bug when batch size = 1.