[Jetson TX2] Yolo 메모리 죽을때(killed)

문제 발생

low memory warning!!

memory available for new process:...

 

TX2보드를 깔고 YOLO 사이트에서 하라는대로 하는데 위 그림과 같이 레이어 64쯤 가서  화면이 멈추고 메모리가 죽는 경우가 있습니다. 아래와 같이 제대로 쳤는데도 말이죠.

/darknet detector demo cfg/coco.data cfg/yolov3.cfg yolov3.weights 동영상

 

이럴땐 Batch size의 문제가 명확합니다.

YOLO를 Git clone하여 받으면 cfg파일이 training시에 사용한 batch size인 64로 되어 있기 때문이죠.

따라서 이를 수정해야 합니다.

 

 

Batch size 수정

cd ~/Works/yolo/darknet/cfg
gedit yolov3.cfg

 

맨 위에 batch와 subdivisions로 주석을 바꿔줍니다.

 

아주 잘 되네요 full network라 느려서 그렇지..ㅠㅠ