DETR Visulization


DETR Visulization

Hook Visualization model output DETR Transformer attention 해당 글은 DETR을 비롯해서 모든 Weigth를 사용하는 모델에서 중간 출력물을 확인하기 위해 조사하였습니다. 여기서는 Transformer 내부의 encoder출력을 확인하면서 시각화하는 작업까지를 확인합니다. 중간에 뇌피셜도 많이 포함되어 있어서 핵심만 파악하면 좋습니다. Deformable DETR은 DETR을 기본적으로 따르는 코드입니다. 아래는 DETR에서 제공하는 중간단 출력확인을 위한 코드입니다. 이것을 사용해서 모델의 중간 출력단을 확인합니다. Google Colaboratory Notebook Run, share, and edit Python notebooks colab.research.google.com # 위에서 설명한 DETR 모델을 불러오기 위한 코드입니다. 메타 측에서 이미 서빙해두었기 때문에 쉽게 사용할 수 있습니다. model = torch.hub....


#attention #Visualization #register_forward_hook #Pytorch #outputhook #jjunsss #hook #forwardhook #featuremap #encoderattention #detrvisualize #detr #attentionweight #attentionVisualize #중간단출력확인

원문링크 : DETR Visulization