[DarkFlow] Yolo.Weight 를 사용해서 Model 학습

(venv) sps@odoaui-MacBookPro project % open darkflow 

 

 

// Darkflow folder 내에  labels.txt file을 수정해준다 

 

 Annotation 작업에 사용했던  Labels 이름을 넣어주면 된다 

 

cfg 파일의 yolo.cfg 파일을  수정해준다 filter class

class 는 label 개수 filters 는 class +5

 

 

 

(venv) sps@odoaui-MacBookPro darkflow % python flow --model ./cfg/TestYolo.cfg --labels ./labels.txt --trainer adam --dataset ./data/ --annotation ./dataset/ --train --summary ./logs --batch 5 --epoch 100 --save 50 --keep 5 --lr 1e-04

 

 

Test

python shell 에서 실행

sudo idle3