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, grid_spacing=10,
arrow_scale=1.0 arrow_scale=1.0
) )
cv2.imshow("形变场 Deformation", arrows) cv2.imshow("Deformation", arrows)
# 步骤 3.2: 在图像上触碰点和区域 # 步骤 3.2: 在图像上触碰点和区域
image_with_foe = orisys.util.draw_contact(img, contour, centroid) 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(): def main():
# Parameters for the pattern. # Parameters for the pattern.
# For example, using 100x100 pixels image with patches of 3x3 pixels. # For example, using 100x100 pixels image with patches of 3x3 pixels.
height = 2000 height = 1000
width = 2000 width = 1000
patch_size = 3 patch_size = 20
# Set r to a value that gives sufficient difference. # Set r to a value that gives sufficient difference.
# Note: r is the threshold for squared difference per channel, and the condition is # 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. # |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 # Initialize HHD solver
if self.__hhd is None: 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 # Force computation
if self.cuda and self.CUPY_AVAILABLE: if self.cuda and self.CUPY_AVAILABLE: