TensorFlow Similarity 0.13.4图像相似度建模工具
图相似度建模的利器就是 tensorflow_similarity-0.13.4.tar.gz 这个包。它不是传统意义上的图像识别库，而是帮你搞定“这个图像像不像另一个图”的问题，像人脸检索、产品推荐场景都能用上，挺好上手的。
TensorFlow 的功能你早就熟得不行了吧？那你一定会觉得 tensorflow_similarity 蛮顺手的，完全是围绕着它打磨出来的。你只要用熟 model.fit()，剩下的流程都比较自然。
它用的是 embedding + 最近邻搜索 的套路，跑完训练后，每张图片都能转成一个向量，用最近邻模型找出最接近的那些。效率还不错，模型压缩后上生产也不费劲。
哦对了，它还集成了像 TripletLoss 和 ContrastiveLoss 这些经典损失函数，不用你自己手写，直接套上就行，效果也蛮稳的。配合 tf.data 用，训练过程也丝滑。
如果你在做图像检索或者想提高搜索精度，这个库真的可以试试。训练和部署都比较轻便，不会让人头大。
顺带推荐几个相关的资源，像 Pandas 数据库 和 Python 数据与可视化实战案例，配合使用，效果更佳。
文件大小：56.09KB
评论区