본문 바로가기

Machine Learning 이야기

(2)
Darknet detector 여러가지 오류 처리하기 Darknet의 detector 예제를 다른 프로그램에 가져다 쓰고 싶어서 원본 폴더에서 필요한 파일들만 복사를 하니 여러가지 오류가 발생했습니다... 열심히 구글검색도 하고, 없는 건 직접 찾아보기도 하면서 처리해 본 오류들을 올려보겠습니다. 부디 Darknet 이용하시는 분들께 도움이 되기를 바랍니다 Darknet의 코드를 분석해 보니 그대로는 다른 코드에 이식이 거의 불가능한 수준이었습니다. 일반적으로 사용한다면 Darknet에서 제공해주는 프로그램을 그대로 사용해서 결과를 출력해 볼 수 있겠습니다만 해당 프레임워크를 이용해 결과를 다른 프로그램에서 처리하도록 만들려다 보니까 문제가 많이 발생하는 코드였습니다. 제가 만났던 오류들을 정리해 보려고 합니다 1. import darknet as dn 오..
pb(frozen graph) 파일에서 input, output node 찾기 darknet 혹은 darkflow를 이용해 만든 yolo weight 파일을 다른 코드에서 이용하기 위해 pb(frozen gragh)파일로 만든 뒤 다른 코드에 적용하려고 시도하던 도중.. input node name과 output node name을 알지 못하면 데이터를 집어넣을 수 없는 현상이 일어났습니다. 여러가지로 찾아본 결과 bazel을 이용해 summarize graph tool을 이용하는 방법이 많이 보였습니다. ...만, pip를 이용해 tensorflow를 설치하는 경우가 많은 요즘 summarize graph tool을 이용하기 위해 tensorflow를 직접 빌드하는 방법은 많이 번거롭다는 생각이 들었습니다. 좀 더 찾아보니 python 코드를 이용해서 간단하게 레이어 구조를 볼 ..