3pp-mirror/Meshroom
1
0
Fork 0
mirror of https://github.com/alicevision/Meshroom.git synced 2025-07-24 03:57:28 +02:00
Code Issues Projects Releases Packages Wiki Activity

[blender] preview script: more robust search for undistorted images

Browse source
This commit is contained in:
Loïc Vital 2023-05-30 11:37:51 +02:00
parent 3a993a9d82
commit d3da5998af
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
meshroom/nodes/blender/scripts/preview.py
View file

@ -143,8 +143,11 @@ def setupRender(view, intrinsic, pose, outputDir):
def setupBackground(view, folderUndistorted):
'''Retrieve undistorted image corresponding to view and use it as background.'''
baseImgName = os.path.splitext(os.path.basename(view['path']))[0]
undistortedImgPath = glob.glob(folderUndistorted + '/*' + baseImgName + "*")[0]
matches = glob.glob(folderUndistorted + '/*' + view['viewId'] + "*") # try with viewId
if len(matches) == 0:
baseImgName = os.path.splitext(os.path.basename(view['path']))[0]
matches = glob.glob(folderUndistorted + '/*' + baseImgName + "*") # try with image name
undistortedImgPath = matches[0]
bpy.ops.image.open(filepath=undistortedImgPath)
undistortedImgName = os.path.basename(undistortedImgPath)
bpy.context.scene.node_tree.nodes["Image"].image = bpy.data.images[undistortedImgName]