()
| 34 | |
| 35 | @memoized |
| 36 | def get_gaussian_map(): |
| 37 | gaussian_map = np.zeros((368, 368), dtype='float32') |
| 38 | for x_p in range(368): |
| 39 | for y_p in range(368): |
| 40 | dist_sq = (x_p - 368 / 2) * (x_p - 368 / 2) + \ |
| 41 | (y_p - 368 / 2) * (y_p - 368 / 2) |
| 42 | exponent = dist_sq / 2.0 / (21**2) |
| 43 | gaussian_map[y_p, x_p] = np.exp(-exponent) |
| 44 | return gaussian_map.reshape((1, 368, 368, 1)) |
| 45 | |
| 46 | |
| 47 | def CPM(image): |