Ошибка: объект numpy.ndarray не может быть вызван
если тип (input_batch) равен np.ndarray: imshow (input_batch [0,:,:, 0], target_batch [0,:,:], wmap_batch [0,:,:])