I have tried to swap multi faces, here is the issue I have encountered. Please share thoughts, if anyone is able to replicate the issue or has fixed it.
Traceback (most recent call last):
File "inference.py", line 153, in <module>
main(args)
File "inference.py", line 104, in main
BS=args.batch_size)
File "/content/drive/MyDrive/sber-swap/utils/inference/core.py", line 63, in model_inference
target_batch_rs = transform_target_to_torch(resized_frs, half=half)
File "/content/drive/MyDrive/sber-swap/utils/inference/core.py", line 18, in transform_target_to_torch
target_batch_rs = target_batch_rs[:, :, :, [2,1,0]]/255.
IndexError: too many indices for tensor of dimension 1
I have tried to swap multi faces, here is the issue I have encountered. Please share thoughts, if anyone is able to replicate the issue or has fixed it.