yanxp / MetaR-CNN

Meta R-CNN : Towards General Solver for Instance-level Low-shot Learning
https://yanxp.github.io/metarcnn.html
177 stars 23 forks source link

the details of "object attentive vector channel-wise product rois" during meta training #15

Open Hxx2048 opened 4 years ago

Hxx2048 commented 4 years ago

Hello! Thank you for releasing code, I have some questions about the details in meta training.

In ./model/faster_rcnn/faster_rcnn.py Line 122,

" proposal_labels = rois_label[b 128:(b + 1) 128].data.cpu().numpy()[0] "

This code means you just choose first label of 128 rois in each batch as the unique label, and you only use object attentive vectors just for this label's class ? Why do you choose only the first one instead of all the labels of 128 rois?
or from your paper, I thought the code should be " proposal_labels = rois_label[b 128:(b + 1) 128].data.cpu().numpy()"

image

yanxp commented 4 years ago

Hello, proposal_labels = rois_label[b 128:(b + 1) 128].data.cpu().numpy()[0] is the first label of 128 rois in each batch, please check it.