search

mangye16 / Cross-Modal-Re-ID-baseline

Pytorch Code for Cross-Modality Person Re-Identification (Visible Thermal/Infrared Re-ID)
MIT License
336 stars 71 forks source link

Update model.py #28

Closed workingcoder closed 5 months ago

workingcoder commented 1 year ago

First of all, thank you for providing this code as a benchmark method for the VI-ReID task.

During the experiment, I found a serious problem in the Non-Local module. The inter_channels of Non_local is defined incorrectly, which makes it always equal to 1 and seriously reduces its ability. We should make the following modification: self.inter_channels = reduc_ratio//reduc_ratio -> self.inter_channels = in_channels//reduc_ratio

Considering that your code is widely used by researchers in this field, hope you can fix this problem as soon as possible.

Best wishes. Happy Spring Festival ! (^-^)

Hua-XC commented 1 year ago

I think your modification is very good