如何使用 keras 演示代碼 siamese_contrastive.py 來使用自定義數據集？

Question

我正在使用具有對比損失的暹羅網絡進行此示例圖像相似性估計。

給定的代碼片段直接從keras.datasets.mnist.load_data()讀取。

我正在嘗試調整此示例並嘗試提供新的數據集。 我有一個目錄，其中保存了正面和負面圖像的示例。

所以我有三個目錄，分別是anchor 、 positive和negative examples ，即：

anchor/
1.jpg
2.jpg
3.jpg
....

positive/
1.jpg
2.jpg
3.jpg
....

negative/
1.jpg
2.jpg
3.jpg
....

我怎么能把這個數據集輸入這個模型？

對於我的情況，我必須在輸入模型之前嵌入圖像。 歡迎任何想法。

Answer 1

您可以像這樣加載圖像：

anchor_images = sorted([str(anchor_images_path / f) for f in os.listdir(anchor_images_path)])

正面和負面都一樣。

詳細示例在這里：

https://keras.io/examples/vision/siamese_network/