Reworked pathfinding in merge_textures

This commit is contained in:
Killergnom
2024-03-24 20:34:20 +01:00
parent 7512e0fdfe
commit b1257a97a4
20 changed files with 21727 additions and 5 deletions

View File

@@ -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)