Question : plt.imshow typeerror invalid dimensions for image data
Answered by : merwan
if image.shape[2] == 1: image = image[:, :, 0]
return image
Source : https://www.google.com/url?sa=t&rct=j&q=&esrc=s&source=web&cd=&cad=rja&uact=8&ved=2ahUKEwiNxdbLytXwAhWV7aQKHWskBv8QFjABegQIAhAD&url=https%3A%2F%2Fdiscuss.pytorch.org%2Ft%2Fhow-to-solve-this-error-invalid-dimensions-for-image-data%2F54430&usg=AOvVaw25qvZNoIIfImj5hK2XJAQe | Last Update : Wed, 26 May 21