fix hhd cuda bug
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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.
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user