from torchvision.utils import save_image
save_image(gen_imgs.data[:25], “images_v1/%d.png” % batches_done, nrow=5, normalize=True)
#gen_imgs.data 【N,C,W,H】
效果:
2、人脸搜索
self.cat = faiss.IndexFlatL2(64) 64为保存的图片矩阵特征长度
self.cat.add(cat_rs) cat_rs为图库的特征pkl文件内容
score, index= index.search(embed_expand, 4) embed_expand 搜索特征,4数量