imdex

IMDEX

IMDEX是一個庫，允許在圖像集上進行語義搜索

如何使用

安裝

要安裝庫，只需在您的

 pip install imdex

模塊

安裝庫後，您只需要導入要使用的模塊，可用的模塊就在下面列出。

 Imdex
  |-Loader
  |-Captioner
  -Indexer

加載程序

加載程序模塊是導入和格式化所有要索引的圖像的簡單方法，它允許您導入單個圖像或文件夾，它可以正確返回圖像，並準備使用大小

 from imdex.loader import load_image, load_foder

images = [load_image("./sample_img/surf.jpg")]#importing single image
images, references = load_foder("sample_img")#importing all images in a folder with their names to be used as references to the image

字幕人

該模塊負責為圖像字幕，由索引模塊調用，但可以獨立導入和使用

 from imdex.Captioner import Captioner

cap = Captioner()

images = []#list of images

cap.captionize(images)# returns a list of captions

索引器

索引器是負責檢索圖像的類，一旦圖像包含在索引器中，字幕模塊將提供英語描述，該描述可以保存為文件並稍後加載，所有圖像都需要一個字符串引用來識別它們，例如它們的名稱。為了查詢圖像，它們的描述和查詢句子被轉換為嵌入的向量，以測量這些距離之間的距離，當計算距離時，將創建排序的向量並返回作為函數的輸出。

 from imdex.Indexer import Indexer

idr = Indexer()

images, references = load_foder("sample_img")

idr.add_images(images, references)

idr.query("a cat is laying")

idr.save_to_csv("images.imdex")

要加載描述，您必須創建索引類類的新實例，並將文件的名稱用作構造函數中的參數

 idr = Indexer(captions_csv_path="images.imdex")

未來的改進

作為一個新庫，它有很多改進要做的改進，這是主要庫的列表：

• 查詢時間優化
• 多語言查詢支持
• 字幕模型的改進
• 在搜索中啟用自定義單詞嵌入模型

想幫忙嗎？將合併請求或電子郵件發送到[email protected]