반응형 잡지식 저장고14 torch.nn.Parameter 에 관해서 Pytorch 에는 Parameter라는 모듈이 있는데, 얘는 레이어가 아니라 말 그대로 파라미터 값만을 가지고 있는 놈이다. class Actor(nn.Module): def __init__(self, num_inputs, num_outputs, continuous=True, shared=False): self.num_inputs = num_inputs self.num_outputs = num_outputs super(Actor, self).__init__() self.fc1 = nn.Linear(num_inputs, hp.hidden) self.fc2 = nn.Linear(hp.hidden, hp.hidden) self.fc3 = nn.Linear(hp.hidden, num_outputs) self... 2021. 4. 7. 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.. 2021. 3. 30. 이전 1 2 3 4 다음
