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..


원문링크 : detectron2 에서 Faster R-CNN RPN에 GradCAM 붙이기