daviddirethucus / face-mask_inpainting Goto Github PK

Hi,

Thank you for your amazing work.
I would like to train the model myself.
Could you please provide the 4k paired images you used to train map module model.
By that, I mean pairs of masked face image, and the ground truth mask binary mask.
It would be a lot of help

Thank you.

Hello ,can you send them foe me?

We are using facenet = torch.load('../Model/InceptionResNetV1_ArcFace.pt') from the shared gDrive

(model-training) C:\Users\ranji\AML-ReferenceCode\Masked_Face_Recognition\Code>python ./main_test.py
Device: cpu
0%| | 0/456 [00:22<?, ?it/s]
Traceback (most recent call last):
File "C:\Users\ranji\AML-ReferenceCode\Masked_Face_Recognition\Code\main_test.py", line 72, in
dist_threshold = evalulate(facenet, eval_loader1, eval_loader2, device)
File "C:\Users\ranji\AML-ReferenceCode\Masked_Face_Recognition\Code\test.py", line 56, in evalulate
dist1 = result(model,eval_loader1, device)
File "C:\Users\ranji\AML-ReferenceCode\Masked_Face_Recognition\Code\test.py", line 42, in result
distance = torch.norm(eval_out - eval_pait_out, dim=1)
TypeError: unsupported operand type(s) for -: 'dict' and 'dict'