Skip to content

Latest commit

 

History

History
81 lines (49 loc) · 4.71 KB

README.md

File metadata and controls

81 lines (49 loc) · 4.71 KB

Person re-IDentification using PyTorch

This repository has been created for the Person Re-Identification project made for the Trends & Applications of Computer Vision UniTN course.

The project aims to adapt a system of Person Re-Identification to Vehicle Re-Identification.

Outline

First milestone Report Final milestone

Introduction

Person Re-Identification (Re-ID) is the task of identifying a specific person across different camera views and locations. It is a critical problem in the field of computer vision and has numerous applications, such as in surveillance systems, smart cities, and retail analytics. The goal of Person Re-ID is to accurately match a person's appearance in one camera view with their appearance in another camera view, even if the person is wearing different clothes or their appearance has changed due to factors such as lighting or pose.

person_ReID

Vehicle Re-Identification (Re-ID) is the task of identifying a specific vehicle across different camera views and locations. It is a problem that is similar to Person Re-ID, but applied to vehicles instead of people.

vehicleReID

NFormer

For this project, we decided to use the repository of the paper NFormer: Robust Person Re-identification with Neighbor Transformer.

In the figure below is reported the architecture of the NFormer:

pipeline

Getting Started 👨‍💻

Training 🚀

Regarding Person Re-ID task: You can run Experiment-all_tricks-tri_center-market.sh to train the Nformer on the Market1501 dataset.

If you want to run the training for the task vehicle Re-ID you have to change the dataset in the file Experiment-all_tricks-tri_center-market.sh from Market1501 to Veri. Then you can run it.

To run the training:

sh Experiment-all_tricks-tri_center-market.sh

Evaluation 🔮

For the evaluation on the Market1501 dataset you can run Test-all_tricks-tri_center-feat_after_bn-cos-market.sh.

If you want to run the evaluation on the Veri dataset you have to change the dataset name in the file Test-all_tricks-tri_center-feat_after_bn-cos-market.sh, moreover you have to change the path of the weights in the folder configs.

To run the evaluation:

sh Test-all_tricks-tri_center-feat_after_bn-cos-market.sh

Dataset

The dataset used for Person Re-Identification is the Market1501 available at this link.

Regarding the Vehicle Re-Identification task, you have to ask for the permission to use the dataset at Veri-776 dataset.

Contacts

Andrea Bonomi - Github - LinkedIn - UniTN Email
Khouloud Ismail - Github - LinkedIn - UniTN Email
Francesco Laiti - Github - LinkedIn - UniTN Email
Davide Lobba - Github - LinkedIn - UniTN Email
Evelyn Turri - Github - LinkedIn - UniTN Email

Acknowledgments

We thank professor Zhun Zhong for providing us the dataset Veri-776.

Furthermore, we thank professor Cecilia Pasquini for providing us a machine for the training of the Nformer model.