Darknet의 weight로 사물인식 테스트

Darknet의 weight로 사물인식 테스트

 

커스텀 데이터셋을 Darknet로 yolov2-tiny로 훈련하여 나만의 weight파일을 얻고 이를 keras를 거쳐 onnx로 컨버팅이 성공적으로 잘되었다.

 

물론 최종적인 unity에서도 인식은 잘 되었다.

그런데 문제가 제대로 된 인식이 되지 않는다. 인식률에 문제가 있긴 해도 아무거나 인식하게 되면 문제가 있는 게 아닌가... 그래서 다시 Darknet로 test를 아래와 같이 진행하였는데, 너무 잘된다..

물론 인식률은 표본이 적어서 문제가 있기는 해도 인식은 되지 않는가...

 

그렇다면 컨버팅 과정이나 unity에서 인식하는 무엇인가가 잘 처리되지 못한다는 것인데... 

현재 이것을 찾고 있는 중이다.  

 

원래 8월까지로 마무리하려했는데 이 부분까지만 해결하고 마무리해야 될 것 같다.

 

캔 인식 - 1

 

종이팩 인식

 

캔 인식 - 2

 

캔 인식 - 3

 

 

 

 

인식 동영상으로 생각보다 잘 되지만 인식률은 좀 낮은 것 같다. 아마도 학습데이터가 적어서 그런 것 같다.