fix(nodes): MLSD needs inputs to be multiples of 64

This commit is contained in:
psychedelicious
2024-09-11 20:07:07 +10:00
committed by Kent Keirsey
parent 7d15f9381d
commit b6aba92426

View File

@@ -48,8 +48,8 @@ class MLSDDetector:
height, width, _channels = np_img.shape
# This model requires the input image to have a resolution that is a multiple of 8
np_img = resize_to_multiple(np_img, 8)
# This model requires the input image to have a resolution that is a multiple of 64
np_img = resize_to_multiple(np_img, 64)
img_output = np.zeros_like(np_img)
with torch.no_grad():