fix: output image type

2023-09-26 16:37:48 +08:00
parent 1a3a559a67
commit b25b0d7552
1 changed files with 1 additions and 1 deletions
@@ -67,7 +67,7 @@ def genreate_mesh_from_texture(bpy, image):
    cv2.drawContours(image, contours, -1, (0, 255, 0), 3)

    # Convert image back to tensor
-    image = [torch.from_numpy(image)]
+    image = torch.from_numpy(np.expand_dims(image, axis=0))
    return (image, meshes[0])