Создать более толстую карту тепла - PullRequest
Новая
       13

Создать более толстую карту тепла

0 голосов
/ 16 января 2019

Привет, я пытаюсь сгенерировать heat map для заданных center точек с singma значением nxm матрицы в python с использованием pytorch.

Я и запускаю следующий код 

import torch
import numpy as np
import math
heat_map = torch.FloatTensor(64, 64).fill_(0)
center_x = 30
center_y = 35
sigma_horz = 0.01953125
sigma_vert = 0.01953125
amplitude = 30
start_time = time.time()
for i in range( hmh):
          for j in range( hmw):
            heat_map[i,j] = amplitude * math.exp(-(math.pow((j-center_x)
                                            /(sigma_horz*hmw),2)/2
                          + math.pow((i-center_y)
                                           /(sigma_vert*hmh),2)/2))
elapsed_time = time.time() - start_time
print (elapsed_time*1000)
plt.imshow(heat_map, interpolation='bilinear'); plt.show()

Для создания тепловой карты требуется 30 micro-sec.

может кто-нибудь сделать это так же быстро, как и вокруг1 micro-sec

Спасибо и привет.

...