/nft-search-model

数据库里有全量的nft图片,用户的query可以从数据库里搜索最相似的nft图片。这个repo可以帮助用户训facenet模型,对每个nft图片产生一个表示向量,在搜索计算相似度的时候用到。

Primary LanguageJupyter NotebookMIT LicenseMIT

使用facenet网络训练模型,可以获得每个nft的向量表示

搜索的时候使用faiss,去全量的nft dataset里面搜索最相似的向量,这个向量对应的nft就是搜索结果。