Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Sakoe chiba band #24

Open
wants to merge 4 commits into
base: master
Choose a base branch
from
Open

Sakoe chiba band #24

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
26a82bd
Sakoe-Chiba band for cpu
toinsson Jul 1, 2022
e9c2654
Sakoe-Chiba band for gpu
toinsson Jul 1, 2022
c2d1868
fix gpu indexing
toinsson Jul 1, 2022
e9c6ad4
Merge branch 'Maghoumi:master' into sakoe-chiba_band
toinsson Jul 5, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 17 additions & 4 deletions soft_dtw_cuda.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -61,8 +61,12 @@ def compute_softdtw_cuda(D, gamma, bandwidth, max_i, max_j, n_passes, R):

# Only compute if element[i, j] is on the current anti-diagonal, and also is within bounds
if I + J == p and (I < max_i and J < max_j):

# Don't compute if outside bandwidth
if not (abs(i - j) > bandwidth > 0):
i_sc, j_sc = i, j
if max_j > max_i: i_sc = i * max_j / max_i
if max_j < max_i: j_sc = j * max_i / max_j
if not (abs(i_sc - j_sc) > bandwidth > 0):
r0 = -R[b, i - 1, j - 1] * inv_gamma
r1 = -R[b, i - 1, j] * inv_gamma
r2 = -R[b, i, j - 1] * inv_gamma
Expand Down Expand Up @@ -101,7 +105,10 @@ def compute_softdtw_backward_cuda(D, R, inv_gamma, bandwidth, max_i, max_j, n_pa
R[k, i, j] = -math.inf

# Don't compute if outside bandwidth
if not (abs(i - j) > bandwidth > 0):
i_sc, j_sc = i, j
if max_j > max_i: i_sc = i * max_j / max_i
if max_j < max_i: j_sc = j * max_i / max_j
if not (abs(i_sc - j_sc) > bandwidth > 0):
a = math.exp((R[k, i + 1, j] - R[k, i, j] - D[k, i + 1, j]) * inv_gamma)
b = math.exp((R[k, i, j + 1] - R[k, i, j] - D[k, i, j + 1]) * inv_gamma)
c = math.exp((R[k, i + 1, j + 1] - R[k, i, j] - D[k, i + 1, j + 1]) * inv_gamma)
Expand Down Expand Up @@ -193,7 +200,10 @@ def compute_softdtw(D, gamma, bandwidth):
for i in range(1, N + 1):

# Check the pruning condition
if 0 < bandwidth < np.abs(i - j):
i_sc, j_sc = i, j
if M > N: i_sc = i * M / N
if M < N: j_sc = j * N / M
if 0 < bandwidth < np.abs(i_sc - j_sc):
continue

r0 = -R[b, i - 1, j - 1] / gamma
Expand Down Expand Up @@ -226,7 +236,10 @@ def compute_softdtw_backward(D_, R, gamma, bandwidth):
R[k, i, j] = -np.inf

# Check the pruning condition
if 0 < bandwidth < np.abs(i - j):
i_sc, j_sc = i, j
if M > N: i_sc = i * M / N
if M < N: j_sc = j * N / M
if 0 < bandwidth < np.abs(i_sc - j_sc):
continue

a0 = (R[k, i + 1, j] - R[k, i, j] - D[k, i + 1, j]) / gamma
Expand Down