Yolov2, v3-tiny 훈련 다시...

congfig에서 변경 안 한 것이 있어서 다시 훈련을 돌렸다.. 

이번 다시 돌리는김에 yolov2-tiny 와 yolov3-tiny를 같이 돌림.. v3 훈련도중 다음과 같은 오류가 발생하여 사이즈를 낮춰서 재실행했더니 정상적으로 훈련이 되었다. 혹시나 v3이 안될 수 있기 때문에 2개를 돌린 것이다.

 

cuDNN isn't found FWD algo for convolution.

 

해결책은 해상도를 낮춰서...  그리고 두 버전의 훈련과정은 좀 다른 것 같다.

Yolo v2 tiny 훈련

 

Yolo v3 tiny 훈련

 

 

물론 따로 돌려도 되는데 GPU 점유율을 생각보다 적게? 차지하는 것 같아서 동시에 돌렸다. 그래서 메모리 문제가 발생된 듯.

 

훈련시간은 서로 다른데 v3이 더 빨랐다. 

전체 훈련 시간의 경우 v2는 약 7~8시간, v3은 6시간 가량 소요되었다.

 

어찌 되었든 잘만 되면 되고 GPU를 돌리는 게 CPU보다 빠르다. (펜티엄 골드 CPU..)

 

 

이제 이 훈련 결과를 다시 keras의 .h5로 변환한 뒤 다시 onnx로 변환했다.

그리고 이것을 Unity에서 아래와 같이 잘 불러오고 정보도 확인되었다. v2와 v3도 정상적으로 불러와졌다.

Unity에서 훈련결과 불러오기

 

이제 다 된것처럼 보이지만... 

실제론 뭐가 문제인지 인식되지 않았다... 아.. 다시 건드려봐야 하나...