fix hhd cuda bug

This commit is contained in:
TPP-G6
2026-01-29 21:24:12 +08:00
parent c3fa34e66d
commit 6f0b6cb9d4
3 changed files with 5 additions and 5 deletions
+1 -1
View File
@@ -61,7 +61,7 @@ def main():
grid_spacing=10,
arrow_scale=1.0
)
cv2.imshow("形变场 Deformation", arrows)
cv2.imshow("Deformation", arrows)
# 步骤 3.2: 在图像上触碰点和区域
image_with_foe = orisys.util.draw_contact(img, contour, centroid)
+3 -3
View File
@@ -115,9 +115,9 @@ def generate_color_pattern(h, w, d, r):
def main():
# Parameters for the pattern.
# For example, using 100x100 pixels image with patches of 3x3 pixels.
height = 2000
width = 2000
patch_size = 3
height = 1000
width = 1000
patch_size = 20
# Set r to a value that gives sufficient difference.
# Note: r is the threshold for squared difference per channel, and the condition is
# |c - c_neighbor| >= sqrt(r). Very high r may make it impossible to satisfy.
+1 -1
View File
@@ -228,7 +228,7 @@ class Sensor:
# Initialize HHD solver
if self.__hhd is None:
self.__hhd = nHHD(cuda=self.cuda, Qgrid=flow.shape[:2], spacings=(18, 18))
self.__hhd = nHHD(cuda=self.cuda, grid=flow.shape[:2], spacings=(18, 18))
# Force computation
if self.cuda and self.CUPY_AVAILABLE: