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