Сложный тип данных изображения подушки?

#deep-learning #python-imaging-library #fft #complex-numbers

Вопрос:

Подводя итог: я импортировал изображение подушки, я преобразовал его в массив np, я выполнил преобразование Фурье (scipy), затем IFFT, а затем я хочу преобразовать его обратно в подушку, чтобы передать его через сеть. (Я использую PyTorch).

Таким образом, мой код:

     def __call__(self, sample):
        image  = sample
        img = np.asarray(image, dtype=np.complex)

        fft_orig = fft(img)
        fftShift = fftshift(fft_orig)
        
        image = Image.fromarray(fftShift.astype(np.complex), 'RGB')
        
        return image
 

Это работает, и это продолжает мое обучение без каких-либо ошибок.

Если я положу туда:

  image = Image.fromarray(fftShift.astype('float32'), 'RGB')

 image = Image.fromarray(fftShift.astype('uint8'), 'RGB')
 

Я получаю предупреждение о том, что код удаляет Воображаемую часть.

Как я могу проверить, может ли изображение подушки (объект) правильно поддерживать сложные типы данных?