Skip to content

😎 Finding duplicate images made easy!

License

Notifications You must be signed in to change notification settings

clibdev/imagededup

Repository files navigation

Differences between original repository and fork:

  • Compatibility with PyTorch >=2.3. (🔥)
  • Converted ONNX models from GitHub releases page. (🔥)
  • The following deprecations has been fixed:
    • SetuptoolsDeprecationWarning: Invalid dash-separated options.
    • FutureWarning: Cython directive 'language_level' not set.

Installation

pip install -r requirements.txt
python setup.py build_ext --inplace

Pretrained models

  • Download links:
Name Model Size (MB) Link SHA-256
MobileNetV3-Small-Dedup 3.7
3.6
PyTorch
ONNX
e04376504b51694ec16ea5473eaf7ba419061953323d89c8e2e3f0bd2087e51b
cfdb10bd95a6b02c1c41618e2d9037d0ca835279c6d8b4d93553a7a04c5ca327

Demo

python demo.py

Export to ONNX format

pip install onnx
python export.py

About

😎 Finding duplicate images made easy!

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages