본문 바로가기

Capstone/2018-2 Capstone

⑤ Yolo_mark

Yolo_mark

리눅스 컴퓨터에 Yolo_mark를 설치합니다.


1. 학습시켜야할 이미지를 Yolo_mark/x64/Release/data/img폴더 안에 넣어줍니다.


2. Yolo_mark/x64/Release/data에 coco.data, coco.names를 넣어줍니다.


3. 다시 Yolo_mark로 돌아가 ./linux_mark.sh를 실행시켜줍니다.


4. 그러면 이미지가 뜨고 class를 줄 부분을 드래그 해주면 됩니다.

이때, Bounding Box를 하면 해당하는 이미지가 있는 train.txt파일이 생성됩니다.

여기에 이미지의 경로가 써있는데 학습할때, darknet으로 옮기면 그에 맞는 절대경로로 바꿔주시면 됩니다.

상대경로로 하니까 자꾸 이미지가 없다고 에러가 발생했습니다.



5. 하다보면 사진이 사용 할 수 없는 경우가 있는데, 이때는 Yolo_mark/x64/Release/data/img 폴더에서 이미지와 txt파일을 삭제하면 됩니다.

   또한 train.txt에도 해당 이미지의 경로를 지워줘야 학습을 할때 이미지가 없다는 에러가 안납니다!!




모든 이미지에 Bounding Box를 하면 img폴더와 coco.data, coco.names, train.txt파일을 darknet에 맞는 폴더에 넣은 후 학습을 시키면 됩니다.


반응형

'Capstone > 2018-2 Capstone' 카테고리의 다른 글

⑥ Yolov3 on raspberry pi 3 B+  (3) 2018.12.23
④ YOLO Custom  (0) 2018.12.14
③ YOLO란?  (0) 2018.12.14
② 이미지 부풀리기  (11) 2018.12.09
① 이미지 크롤링  (0) 2018.12.01