This project performs Person Re-Identification (Re-ID) using deep feature embeddings extracted via a pretrained EfficientNet-B0 model. The system retrieves visually similar identities from a gallery using cosine similarity on learned embeddings.
We used the Market-1501 dataset which consists of:
- 751 training identities
- 750 test identities
- Images captured from 6 camera views
- Backbone: EfficientNet-B0
- Embedding Size: 512-dimensional vector
- Loss Function: Triplet Loss (using hard negative mining)
- Similarity Metric: Cosine similarity
- Training Epochs: 20+
- Framework: PyTorch
- Each image is passed through a pretrained EfficientNet to extract a 512-d embedding.
- A query image is compared to the embeddings of all gallery images using cosine similarity.
- Top-K most similar images are returned as the result.
- A Gradio web interface allows user interaction for live querying.