有什么 python 的包可以直接生成图片的向量吗?用于以图搜图,比较图片的相似度?
有什么封装好的 python 的包,可以直接生成图片的向量吗?
有什么封装好的 python 的包,可以开箱即用,直接生成图片的向量吗?
有什么封装好的 python 的包,可以通过 resnet,残差神经网络开箱即用,直接生成图片的向量吗?
当然有了,看这个:https://github.com/ponponon/i...
from pathlib import Path
from typing import List
from iv import ResNet, l2
# Initialize a residual neural network
resnet: ResNet = ResNet(
weight_file='weight/gl18-tl-resnet50-gem-w-83fdc30.pth'
)
# Generate a vector of specified images
# The generated vector is a List[float] data structure,
# the length of the list is 512, which means the vector is of 512 dimensions
vector_1: List[float] = resnet.gen_vector('example-1.jpg')
vector_2: List[float] = resnet.gen_vector('example-2.jpg')
# Compare the Euclidean distance of two vectors
distance: float = l2(vector_1, vector_2)
print('Euclidean Distance is ', distance)
非常简单,开箱即用
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。