I directly input the UV texture image, how should I modify the code?

[thaoshibe]

Hi, you should modify the code from Line34

Line 34-36: Get UV texture image of imageA and imageB. If you want to directly input the UV texture, please comment L34-36.

[goldwater668]

@thaoshibe Run according to your above operation, report the following error： Traceback (most recent call last): File "main.py", line 58, in output = model.render_texture(new_txt) File "/data/hjq/CPM/makeup.py", line 132, in render_texture self.triangles.T, AttributeError: 'Makeup' object has no attribute 'triangles'

[thaoshibe]

Hi @seawater668 , I'm not sure what you're trying to get. As far as I understand, your inputs are UW-maps, and you want to get outputs as UV maps as well. Then, please comment Line 47, 48 too.

Sorry I can't assist you further. But if you want to modify anything, starting debug from here is a good start. Thank youu.

[goldwater668]

@thaoshibe if name == "main": args = get_args() model = Makeup(args)

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

[thaoshibe]

As mentioned above: Sorry I can't assist you further. But if you want to modify anything, starting debug from here is a good start. Thank you!.