Reworked pathfinding in merge_textures
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
import cv2
|
||||
import sys
|
||||
import numpy as np
|
||||
import re
|
||||
import os
|
||||
|
||||
if len(sys.argv) < 2:
|
||||
print("Please provide the input files.")
|
||||
@@ -36,8 +36,17 @@ aob, metallicb, roughnessb, = orm.T
|
||||
BCR = np.array([basecolorb, basecolorg, basecolorr, roughnessb])
|
||||
NMO = np.array([metallicb, normalg, normalr, aob])
|
||||
|
||||
result = re.search(r'TX_(.*)_', basecolor_file)
|
||||
output_dir = re.search(r'^(.+)\\.+$', basecolor_file).group(1)
|
||||
#result = re.search(r'TX_(.*)_', basecolor_file)
|
||||
output_filename = os.path.basename(basecolor_file)
|
||||
output_filename = output_filename.removeprefix("TX_")
|
||||
output_filename = output_filename.removesuffix("_ALB.PNG")
|
||||
|
||||
cv2.imwrite(output_dir + f"\\{result.group(1)}_BCR.png", BCR.T)
|
||||
cv2.imwrite(output_dir + f"\\{result.group(1)}_NMO.png", NMO.T)
|
||||
if not os.path.exists(os.path.join(os.path.dirname(basecolor_file),"merged")):
|
||||
os.mkdir(os.path.join(os.path.dirname(basecolor_file),"merged"))
|
||||
|
||||
output_dir = os.path.join(os.path.dirname(basecolor_file),"merged")
|
||||
|
||||
#output_dir = re.search(r'^(.+)\\.+$', basecolor_file).group(1)
|
||||
|
||||
cv2.imwrite(output_dir + f"\\{output_filename}_BCR.png", BCR.T)
|
||||
cv2.imwrite(output_dir + f"\\{output_filename}_NMO.png", NMO.T)
|
||||
Reference in New Issue
Block a user