ezhdeha/image-deduplication-model
0
Image Deduplication Model with FAISS
This project detects whether an image is a duplicate of one seen during training on CIFAR-100.
Files
resnet18_cifar100.pth: Trained modelfaiss_index.index: FAISS L2 indexthreshold.txt: Suggested similarity thresholdinference.py: Inference logicapp.py: Gradio app for testing
Usage
from inference import is_duplicate
with open("image.jpg", "rb") as f:
result = is_duplicate(f.read())
print(result)