GNAYUOHZ / ReID-MGN

Simple pytorch unofficial implement of paper: Learning Discriminative Features with Multiple Granularities for Person Re-Identification
235 stars 61 forks source link

Code modification suggestion #38

Open jiollos opened 1 year ago

jiollos commented 1 year ago

https://github.com/GNAYUOHZ/ReID-MGN/blob/f0251e9e6003ec6f2c3fbc8ce5741d21436c20cf/data.py#L81C9-L81C59

In "data.py", line 81, "return int(filepath.split('/')[-1].split('')[0])" is wrong. If you want to get the id, this line should be changed to "return int(filepath.split('/')[-1].split('')[2].split('\')[-1])".