Skip to content

Espadan/Person-reID-CNN_Self-Attention-baseline-pytorch

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

45 Commits

docs

docs

 
 

README.md

README.md

 
 

data_preparation.ipynb

data_preparation.ipynb

 
 

model_result_1_conv.ckpt

model_result_1_conv.ckpt

 
 

model_result_2_conv_bn.ckpt

model_result_2_conv_bn.ckpt

 
 

model_result_3_conv_bn_sa.ckpt

model_result_3_conv_bn_sa.ckpt

 
 

model_result_4_conv_bn_sa_bn.ckpt

model_result_4_conv_bn_sa_bn.ckpt

 
 

train-and-test.ipynb

train-and-test.ipynb

 
 

Repository files navigation

Person-reID-CNN_Self-Attention-baseline-pytorch

The architecture of CNN_SA-baseline:

alt text

Self-Attention layer:

alt text

Example:

alt text

Results:

Accuracy/Epoch:

alt text

Loss/Epoch:

alt text

mAP/Epoch:

alt text

time(sec)/40epoch:

alt text

About

Self Attention layer for Person ReID. Baseline. PyTorch.

Resources

Readme
Activity

Stars

11 stars

Watchers

1 watching

Forks

5 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages