(kernel_size=31, std=0.5)
| 243 | |
| 244 | |
| 245 | def get_gaussian_kernel(kernel_size=31, std=0.5): |
| 246 | n = np.zeros([kernel_size, kernel_size]) |
| 247 | n[kernel_size//2, kernel_size//2] = 1 |
| 248 | k = scipy.ndimage.gaussian_filter(n, sigma=std) |
| 249 | k = k.astype(np.float32) |
| 250 | return k |
| 251 | |
| 252 | |
| 253 | def init_kernel_torch(kernel, device="cuda:0"): |
nothing calls this directly
no outgoing calls
no test coverage detected