arcfaceproblem

arcface

# get embedding for each face
    embeddings = np.zeros([len(loader.dataset), embedding_size])
    with torch.no_grad():
        for idx, (image, _) in enumerate(
            tqdm(loader, desc="Create embeddings matrix", total=len(loader)),
        ):
            print(idx)
            embeddings[idx, :] = F.normalize(backbone(image.to(device))).cpu()
            tensor_to_pil = transforms.ToPILImage()(image.squeeze_(0)).convert('RGB')
            tensor_to_pil.show()