Darknet Yolov3-tiny 훈련 완료

Yolov3-tiny 훈련 완료

지난번 darknet.exe를 빌드하고 nvidia 그래픽카드 드라이버를 업데이트 한 이후 훈련을 진행했다.

 

이번에는 yolo v2가 아닌 v3의 tiny로 진행했다. 모바일에서 사용하기에는 용량이 너무 커서 tiny로 진행했다.

 

물론 데이터 경로문제가 조금 있었지만 금방 해결되었다.

절대경로의 위치가 지난번 darknet_no_gpu때와 달라서 인식 오류가 있었다. 이거 상황에 따라 바뀌는 건지는 모르겠다.

 

train.txt 내의 절대경로가 darknet 실행파일 이하의 경로로 되어 있어야 한다.

 

아무튼 훈련을 시작하고 약 6.2시간만에 아래와 같이 끝났다.

훈련 완료

 

지난번 GPU를 사용하지 않았을때는 무려 5일의 시간이 소요되었다. 차이가 엄청나다..(펜티엄골드 지못미)

 

이제 이 가중치(.weights) 파일을 다시 keras의 .h로 변환 후 onnx로 컨버팅 해서 시도해 봐야겠다.

 

잘되면 데이터셋을 추가로 좀 더 한 다음 재정비한 후 최종 적용해 봐야겠다.

 

근데... 뭔 가중치 파일이 이렇게 많지.....