GithubHelp home page GithubHelp logo

doubtedsteam / mpanet Goto Github PK

View Code? Open in Web Editor NEW
59.0 59.0 5.0 342 KB

The official implementation for Discover Cross-Modality Nuances for Visible-Infrared Person Re-Identification (CVPR21)

License: MIT License

Python 100.00%
person-reidentification pytorch visible-infrared

mpanet's People

Contributors

doubtedsteam avatar someauthor avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar

mpanet's Issues

visualization

Hello, excuse me, I'm very interested in this visual part of the code, can you share it? Thank you

No such file or directory: './SYSU-MM01/exp/rand_perm_cam.mat'

1
您好,我按照您的代码训练时,报了以下错误:
FileNotFoundError: [Errno 2] No such file or directory: './SYSU-MM01/exp/rand_perm_cam.mat'
SYSU-MM01数据集的exp文件夹下没有rand_perm_cam.mat文件,我该怎么办呢?

Issue about addition of the 'v_cls_loss' and 'i_cls_loss'

In the file 'MPANet/models/baseline.py', the 'train_forward' function calculates the final loss composed of multiple components. In the following code snippet, I notice that 'v_cls_loss' and 'i_cls_loss' are added twice.

        if self.mutual_learning:
            # cam_ids = kwargs.get('cam_ids')
            # sub = (cam_ids == 3) + (cam_ids == 6)
            
            logits_v = self.visible_classifier(feat[sub == 0])
            v_cls_loss = self.id_loss(logits_v.float(), labels[sub == 0])
            **loss += v_cls_loss * self.weight_sid**
            logits_i = self.infrared_classifier(feat[sub == 1])
            i_cls_loss = self.id_loss(logits_i.float(), labels[sub == 1])
            **loss += i_cls_loss * self.weight_sid**

            logits_m = torch.cat([logits_v, logits_i], 0).float()
            with torch.no_grad():
                self.infrared_classifier_.weight.data = self.infrared_classifier_.weight.data * (1 - self.update_rate) \
                                                 + self.infrared_classifier.weight.data * self.update_rate
                self.visible_classifier_.weight.data = self.visible_classifier_.weight.data * (1 - self.update_rate) \
                                                 + self.visible_classifier.weight.data * self.update_rate

                logits_v_ = self.infrared_classifier_(feat[sub == 0])
                logits_i_ = self.visible_classifier_(feat[sub == 1])

                logits_m_ = torch.cat([logits_v_, logits_i_], 0).float()
            logits_m = F.softmax(logits_m, 1)
            logits_m_ = F.log_softmax(logits_m_, 1)
            mod_loss = self.KLDivLoss(logits_m_, logits_m) 

            **loss += mod_loss * self.weight_KL + (v_cls_loss + i_cls_loss) * self.weight_sid**
            metric.update({'ce-v': v_cls_loss.data})
            metric.update({'ce-i': i_cls_loss.data})
            metric.update({'KL': mod_loss.data})

Did you do it on purpose with double 'self.weight_sid'?

About TSNE

Hello,where can i get the features without dual path for TSNE ,like ‘ features/sysu/query-sysu-test-nodual-nore-adam-16x8-grey_model_150.mat’

Visualization?

I am interested in the visualization of learned features in this paper.
image

Could you shared the code with me? Thank you!

Rank-1 Train result are stuck at 66%

I have tried multiple trainings with MPANet
But it is consistently getting about 66% of Rank-1 in single shot and all mode of sysu dataset

Is there any method to reproduce the Rank-1 accuracy(70.58%) that is written in your MPANet paper?

by the way this is my environment settings
torch == 1.13.0
torchvision == 0.14.0
ignite == 1.1.0
apex == 0.1

about Fig.3

How to get the visualization result like Fig.3?

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.