seungriyou/wbb_Darknet

2020 NET 챌린지 시즌 7 왕밤빵 - Darknet 수정 코드

wbb_Darknet

NET 챌린지 시즌 7 왕밤빵 - Darknet 수정 코드

설명

• 실시간 물체 인식을 위해 YOLOv3를 이용하여 택배 상자 이미지를 학습하였다.
• Darknet의 소스코드를 수정하여 인식된 택배 상자의 개수가 추출되게 하였고, 연속적인 물체 인식이 한 번의 weight upload만으로 가능하도록 하였다.
• Jetson Nano(Ubuntu 18.04)에서 실행된다.

수정 사항

Original: AlexeyAB/darknet

• Main Modification in src
  • darknet.c: LINK
  • detector.c: LINK
  • image.c: LINK
  • image.h: LINK
• Main Modification in data
  • obj.names: LINK
• Main Modification in cfg
  • obj.data: LINK
  • yolov3_box.cfg: LINK
• There should be backup directory which contains .weights files, but those files are so large that they cannot be uploaded.
  • .weights files were created by learning labelled images of 'box' in HPC & Darknet.

실행 방법

darknet detect cfg/yolov3_box.cfg backup/yolov3_box_5000.weights

인식 예시

전체 동작