SW 연구실
Computer Vision, Deep Learning, 알고리즘
2016년 8월 5일 금요일
darknet train time image resize speed issue
darknet yolo train을 할때, image size나 net 구조, gpu, cpu 사양에 따라 다르겠지만, 내 환경에서는 아래처럼 gpu 가 idle 되는 시간이 많이 발생 했다.
profiling 을 해보니 image를 random 하게 resize 하는 부분에서 cpu 처리가 오래 걸리는 것이 었다.
resize 부분을 opencv +ipp +gpu 로 바꾸고 나서 아래처럼 개선 되었다.
내 환경에서는 4배정도 빨리 train되고 있다.
수정된 code :
https://github.com/ttagu99/darknet
댓글 없음:
댓글 쓰기
최근 게시물
이전 게시물
홈
피드 구독하기:
댓글 (Atom)
댓글 없음:
댓글 쓰기