로딩
티스토리 데이터 처리 중입니다.

detectron2 에서 Faster R-CNN RPN에 GradCAM 붙이기

 detectron2 에서 Faster R-CNN RPN에 GradCAM 붙이기

기본적인 코드 세팅은 github.com/yizt/Grad-CAM.pytorch 를 참조한다. 이 외에 detectron2 설치폴더 밑 modeling/proposal_generator/rpn.py 에서 RPN 클래스 하부 predict_proposals 함수에서 def predict_proposals( self, anchors: List[Boxes], pred_objectness_logits: List[torch.Tensor], pred_anchor_deltas: List[torch.Tensor], image_sizes: List[Tuple[int, int]], ): """ Decode all the predicted box regression deltas to proposals.

Find the to.....