[AI] yolov3 model

snusun·2021년 8월 12일
0

Plip 개발일지

목록 보기
11/25
post-thumbnail

드디어 AI model 개발을 위해 yolo model을 탐색하는 시간이 왔다!

먼저 ssh를 이용해 대회용 remote server에 접속한 다음
darknet folder에 들어가서

./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

를 입력해 detector를 실행시키려니 error 발생!

cm.gpu q=cm_s_vgpu gpu=1

로 gpu 환경을 setting해야한다.

그리고 또 맨 위 명령어를 실행시켰더니 이번에는

Error: in the file data/coco.names number of names 80 that isn't equal to classes=60 in the file cfg/yolov3.cfg
(predictions:70): Gtk-WARNING **: 21:48:46.827: cannot open display: 0:0

이 에러와 warning이 발생했다.

튜토리얼 영상 34분 즈음에 traning하는 방법이 나와있다!
거기서 시키는대로 하면 됨..
대신 이미지가 안나오는 이슈가 있는데
https://147.46.121.38:8443/etx/
이 프로그램을 이용해서 terminal을 켜서 여기서 돌려본 결과

우와 ~ 잘돌아간다

내일 서리에게 모델 알려줘야하니까 절차를 간단히 정리해보자면
1. server 접속 -> gpu 환경 접속
2. Exceed turbox로 terminal 열기
3. darknet folder에서 ./darknet detector train yolov3.data cfg/yolov3.cfg 실행

yolo에 대한 설명은 https://pjreddie.com/darknet/yolo/ 여기!

profile
대학생 근데 이제 컴공을 곁들인

0개의 댓글