2017-04-24 10 views
-1

나는 물체 감지를위한 욜로 아키텍처를 사용하고 싶습니다 완료. 내 사용자 지정 데이터로 네트워크를 교육하기 전에 Pascal VOC 데이터에서이 단계를 수행했습니다. https://pjreddie.com/darknet/yolo/교육 다크 즉시

지침은 매우 명확합니다. 그러나 최종 단계 이후

./darknet detector train cfg/voc.data cfg/yolo-voc.cfg darknet19_448.conv.23

다크 즉시 훈련을 중지하고 무게가 backups/ 디렉토리에 기록되었다고 발표했다.

처음에는 사전 훈련이 너무 좋았고 한 번에 중지 기준에 도달 할 것이라고 생각했습니다. 그래서 나는 시험 이미지 data/dog 중 하나에이 무게로 ./darknet detect 명령을 사용했습니다. 아무것도 발견되지 않았습니다.

사전 웨이트를 사용하지 않으면 네트워크가 학습합니다. 내가 편집 한 CFG/욜로-voc.cfg는

# Testing 
#batch=1 
#subdivisions=1 
# Training 
batch=32 
subdivisions=8 

지금 훈련 과정은 많은 시간 동안 실행 해되었으며 내 GPU를 따뜻하게 유지하고 사용할 수 있습니다.

이 다크을 훈련하는 의도 된 방법이 있나요? 훈련을 끝내지 않고도 예비 가중치를 올바르게 사용할 수있는 방법은 무엇입니까?

체크 포인트를 만들거나 진행 상황을 파악할 수있는 설정이 있습니까?

답변

0

이것은 이전 질문이므로 지금 답변을 보내 주시면 감사하겠습니다.

약 1 개월 동안 darknet으로 작업 한 후 사람들이 포럼에서 요청/게시 한 대부분의 장애물을 만났습니다. 귀하의 경우에는, 나는 가중치가 이미 배치의 최대 수에 대한 훈련을받은 그것 때문에 꽤 확신하고 사전 교육을받은 가중치 다크 가정 교육 읽을 때 이루어졌다.

관련 개인 경험 : 사전 가중치 파일 중 하나를 사용했을 때, 반복 40101에서 시작하여 절단하기 전에 40200까지 실행되었습니다.

사용자 정의 데이터가있는 경우 처음부터 교육해야하지만 사전 학습 된 가중치를 다시 시도하려면 cfg 파일에서 최대 일괄 처리 변경이 도움이 될 수 있습니다. 처음부터 훈련하면

+0

, 어떻게 다른 클래스를 추가/재교육? 임 재 훈련 문제가 있습니다. –