본문 바로가기
TIL

[23.03.30] HRNet Lite

by rahites 2023. 3. 31.

# 미세먼지가 심해서 목이 너무 아프다.. 몸이 너무 안좋음..

 

1. HRNet Lite

오늘은 첫 인턴 출근 날이었다. 물론 기존에도 일은 계속 진행했었기 때문에 다를 바 없었지만, 그래도 회사에 필요한 일을 앞으로 하게 될 것이라는 마음가짐을 잡는 정도로 생각했다. 오늘은 HRNet Lite 코드를 위주로 확인하였는데 참고한 코드는 다음과 같다.

https://github.com/HRNet/Lite-HRNet

 

GitHub - HRNet/Lite-HRNet: This is an official pytorch implementation of Lite-HRNet: A Lightweight High-Resolution Network.

This is an official pytorch implementation of Lite-HRNet: A Lightweight High-Resolution Network. - GitHub - HRNet/Lite-HRNet: This is an official pytorch implementation of Lite-HRNet: A Lightweigh...

github.com

 

Linux에서만 실행이 가능하다 하여 내 실험 환경인 Ubuntu에서 코딩을 진행하였고, Read me를 따라 환경설정을 구축하였다. 하지만 역시나 한번에 쉽게 되는 일은 없었고... 구글링을 통해 환경을 계속 수정해 가며 진행하였다. 그 중 가장 도움을 많이 받은 블로그는 아래 블로그이다.

https://velog.io/@shshin/LiteHRNet-%EC%BD%94%EB%93%9C-%EC%98%A4%EB%A5%98-%EA%B3%A0%EC%B9%98%EA%B8%B0

 

LiteHRNet 코드 오류 고치기

LiteHRNet github 코드를 바탕으로 학습을 하며 생긴 모든 오류들에 대해 작성해보고자 한다.먼저 이해가 되지 않았던 부분에 대해서 정리해보자!처음에 python tools/train.py로 학습을 시작하는 방법이

velog.io

 

정말 내가 났던 오류들의 해결방법을 모두 정리 해두어 위 블로그의 해결방법을 따라가니 학습까지 코드를 진행할 수 있었다!! 학습까지 시간이 며칠 걸리기 때문에 (COCO 데이터 사용) 다음 출근 때 결과를 확인해 볼 생각이다.

 

여담으로 기존에 진행했던 Transformer 코드도 오늘 마무리 하였다! 

'TIL' 카테고리의 다른 글

[23.04.05] Factorization Machines  (0) 2023.04.06
[23.03.29] Transformer와 추천시스템  (0) 2023.03.30
[23.03.28] Wandb와 ONNX  (0) 2023.03.28

댓글