I directly input the UV texture image, how should I modify the code? about cpm HOT 5 CLOSED

imgA = np.array(Image.open(args.input))
imgB = np.array(Image.open(args.style))
imgA = cv2.resize(imgA, (256, 256))
imgB = cv2.resize(imgB, (256, 256))
A_txt = imgA.copy()
B_txt = imgB.copy()
# model.prn_process(imgA)
# A_txt = model.get_texture()
# B_txt = model.prn_process_target(imgB)
if args.color_only:
    output = color_makeup(A_txt, B_txt, args.alpha)
elif args.pattern_only:
    output = pattern_makeup(A_txt, B_txt)
else:
    color_txt = model.makeup(A_txt, B_txt) * 255

    cv2.imwrite('./0929/color_txt.jpg', color_txt)
    mask = model.get_mask(B_txt)
    mask = (mask > 0.001).astype("uint8")
    cv2.imwrite('./0929/mask.jpg', mask)
    new_txt = color_txt * (1 - mask)[:, :, np.newaxis] + B_txt * mask[:, :, np.newaxis]
    cv2.imwrite('./0929/new_txt.jpg', new_txt)
    output = model.render_texture(new_txt)
    output = model.blend_imgs(model.face, output, alpha=1)

x2, y2, x1, y1 = model.location_to_crop()
output = np.concatenate([imgB[x2:], model.face[x2:], output[x2:]], axis=1)
if not os.path.exists(args.savedir):
    os.makedirs(args.savedir)
save_path = os.path.join(args.savedir, "result.png")

Image.fromarray((output).astype("uint8")).save(save_path)
print("Completed 👍 Please check result in: {}".format(save_path))