Tag: pytorch
All the articles with the tag "pytorch".
-
이미지 분류 모델 학습 파이프라인 — 11단계 체크리스트
어제 NDT 실습을 통해 익힌 학습 흐름을 다시 추상화해서, 어떤 이미지 분류 문제에도 적용할 수 있는 단계별 체크리스트로 정리했다. 정규화가 왜 필요한지 같은 작은 의문도 함께.
-
NDT 결함 분류 실습 — PyTorch + ResNet18 전이학습부터 Grad-CAM까지
임베디드 환경에서 객체 인식을 했던 경험을 살릴 기회가 생겼고, 그 김에 NDT(비파괴 검사) 도메인을 다시 잡았다. NEU Steel Defect 데이터셋으로 결함 6종 분류를 전이학습으로 학습시키고 Grad-CAM으로 모델 판단 근거까지 시각화한 실습 기록.