- Implemented dataset and augmentation pipeline in `dataset.py` for handling image pairs and heavy augmentations. - Created a Siamese Network architecture in `model.py` with shared weights and contrastive loss functions. - Developed training script in `train.py` to facilitate model training with validation and early stopping. - Added inference capabilities in `inference.py` for comparing images and finding matches. - Included a perceptual hashing utility in `p_hash.py` for quick image similarity checks. |
||
|---|---|---|
| README.md | ||
| dataset.py | ||
| inference.py | ||
| model.py | ||
| p_hash.py | ||
| train.py | ||