yolo-v2 수정

하나.소스 코드 수정


cfg/coco.data
classes= 1     # 1
train  = /home/pjreddie/data/coco/trainvalno5k.txt
valid  = coco_testdev
#valid = data/coco_val_5k.list
names = data/coco.names
backup = /home/pjreddie/backup/
eval=coco

src/detector.c
void run_detector(int argc, char **argv)// , .line=542
int classes = option_find_int(options, "classes", 1);// 1

void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh) 
draw_detections(im, l.w*l.h*l.n, thresh, boxes, probs, names, alphabet, 1); // 1。line=481

둘.재컴파일 및 테스트


1. 재컴파일
make clean
make

2. 테스트
./darknet detector demo cfg/coco.data cfg/yolo.cfg yolo.weights 

좋은 웹페이지 즐겨찾기