如何使用 resnet 生成圖片向量？

ponponon發表於2023-01-23

有什麼 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)

非常簡單，開箱即用

相關文章