Сюжет Матплотлиба Фигура к тенору Пайторха - PullRequest
Новая
       6

Сюжет Матплотлиба Фигура к тенору Пайторха

0 голосов
/ 04 марта 2020

Мне нужно преобразовать полученное изображение графика matplotlib в тензор.

Теперь я использую эту функцию, которая сначала преобразует изображение numpy массив: 

from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
def fig_to_numpy(fig):
    canvas = FigureCanvas(fig)
    canvas.draw()
    numpy_image = np.frombuffer(canvas.tostring_rgb(), dtype='uint8').reshape(512, 512, 3)
    return numpy_image

Таким образом, я могу затем создать из него тензор 

from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas

fig = plt.figure() .. # Assume I plot things
data_numpy = fig_to_numpy(fig)
data_tensor = torch.from_numpy(data_numpy).float()

Как я могу направить преобразование в тензор Pytorch?

...