search

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

Error? #27

Closed GuHongyang closed 4 years ago

GuHongyang commented 4 years ago

Why does your model have only one reduction? Reductions between branches in the paper are not shared.

    fg_p1 = self.reduction(zg_p1).squeeze(dim=3).squeeze(dim=2)
    fg_p2 = self.reduction(zg_p2).squeeze(dim=3).squeeze(dim=2)
    fg_p3 = self.reduction(zg_p3).squeeze(dim=3).squeeze(dim=2)
    f0_p2 = self.reduction(z0_p2).squeeze(dim=3).squeeze(dim=2)
    f1_p2 = self.reduction(z1_p2).squeeze(dim=3).squeeze(dim=2)
    f0_p3 = self.reduction(z0_p3).squeeze(dim=3).squeeze(dim=2)
    f1_p3 = self.reduction(z1_p3).squeeze(dim=3).squeeze(dim=2)
    f2_p3 = self.reduction(z2_p3).squeeze(dim=3).squeeze(dim=2)