NFormer
NFormer copied to clipboard
haochenheheda
Implementation of NFormer: Robust Person Re-identification with Neighbor Transformer
When I train to the nformer stage, I will be prompted that the video memory is not enough. My video card is 12g video memory. Is there any way for...
File "/media/data4/qiujy/anaconda3/envs/pytorch/lib/python3.9/site-packages/ignite/engine/engine.py", line 291, in _handle_exception raise e File "/media/data4/qiujy/anaconda3/envs/pytorch/lib/python3.9/site-packages/ignite/engine/engine.py", line 272, in _run_once_on_dataset self.state.output = self._process_function(self, batch) File "/media/data4/qiujy/reid/NFormer/./engine/trainer.py", line 119, in _update loss = nformer_loss_fn(score, feat, target) File...
Hello, teacher! I recently read your article "NFormer: Robust Person Re-identification with Neighbor Transformer" and found it very inspiring. I'm very interested in it. However, there's one thing that I...
In your paper, NFormer can be easily applied to other models. How can it be applied to other models? thank you
``` def __getitem__(self, index): ''' 从backbone提取的特征中 ''' center_index = random.randint(0, self.data_num - 1) center_feat = self.feats[center_index].unsqueeze(0) # 1, dim center_pid = self.ids[center_index] selected_flags = torch.zeros(self.data_num) selected_flags[center_index] = 1 # distmat...
I find that the code only provides NFormer with Res50. However, the best performance in your paper is with ABDNet. Could you please tell me how to reproduce with ABDNet?
Metadata
Owner
Metadata
Implementation of NFormer: Robust Person Re-identification with Neighbor Transformer