Fork of idealo/imagededup
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.
pip install -r requirements.txt
python setup.py build_ext --inplace
- Download links:
Name | Model Size (MB) | Link | SHA-256 |
---|---|---|---|
MobileNetV3-Small-Dedup | 3.7 3.6 |
PyTorch ONNX |
e04376504b51694ec16ea5473eaf7ba419061953323d89c8e2e3f0bd2087e51b cfdb10bd95a6b02c1c41618e2d9037d0ca835279c6d8b4d93553a7a04c5ca327 |
python demo.py
pip install onnx
python export.py