profile
컴공생
post-thumbnail

PyTorch Lightning에서 CAM 시각화 구현

저가 못 찾은 건지 모르겠지만 pytorch lightning으로 cam을 만드는 코드를 찾을 수 없었니다. 물론 pytorch로 쉽게 구현할 수 있지만 pl로 만든 모델이라 pytorch로 포팅하는 것 보다 predict step이나 test step에 넣어서 보는게 더 편하다고 생각했습니다. CAM(class activation maps) CAM 논문 CAM은 워낙 잘 설명 되어있는 블로그나 영상이 많아서 패스하겠습니다 PL code 데이터는 데이콘의 도배하자 분류 유형 대회 데이터 셋을 사용했습니다. 코드를 고치기 쉽도록 좀 분리해서 작성했습니다. 코드를 다 적기에는 너무 길어 일부

2023년 5월 23일
·
0개의 댓글
·