yolov 3は自分で訓練したweigtを利用して画像を一括テストし、定義されたフォルダの下に保存する.
yolov 3が持参したテストコマンド知能を利用してdataの下の指定画像をテストし、毎回1枚の画像しかテストできない.
リファレンスhttps://blog.csdn.net/mieleizhi0522/article/details/79989754その後、ブロガーが*GetFilename(char*p)関数を追加したときにエラーが発生し、自分の環境では「charchar」タイプを認識できないことに気づきました.
ブロガーのコードを自分のdetecにコピーします.c後、自分のdetectに対して.c GetFilename(char*p)の機能を実現
1.char fn[30],*p,*qを宣言する.
2.strcpy(fn,(p=strrchr(path,'/'))を追加しますか?p+1 : path);
3.従来のGetFilename(path)の代わりにfnを使用すると、以上の機能が実現できます(再makeを忘れないでください)
コードを追加するには、次のようにします.(3つのコードを変更したり、自分のコンピュータの名前に変えたり、画像が存在するパスを選択したりする必要があります)
実行./darknet detector test cfg/voc.data cfg/yolov3-voc.cfg backup/yolov3-voc_final.Weights後に一括画像のテストを実現
リファレンスhttps://blog.csdn.net/mieleizhi0522/article/details/79989754その後、ブロガーが*GetFilename(char*p)関数を追加したときにエラーが発生し、自分の環境では「charchar」タイプを認識できないことに気づきました.
ブロガーのコードを自分のdetecにコピーします.c後、自分のdetectに対して.c GetFilename(char*p)の機能を実現
1.char fn[30],*p,*qを宣言する.
2.strcpy(fn,(p=strrchr(path,'/'))を追加しますか?p+1 : path);
3.従来のGetFilename(path)の代わりにfnを使用すると、以上の機能が実現できます(再makeを忘れないでください)
コードを追加するには、次のようにします.(3つのコードを変更したり、自分のコンピュータの名前に変えたり、画像が存在するパスを選択したりする必要があります)
void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh, float hier_thresh, char *outfile, int fullscreen)
{
int person_num=0;
list *options = read_data_cfg(datacfg);
char *name_list = option_find_str(options, "names", "data/names.list");
char **names = get_labels(name_list);
char fn[30],*p,*q;
image **alphabet = load_alphabet();
network *net = load_network(cfgfile, weightfile, 0);
set_batch_network(net, 1);
srand(2222222);
double time;
char buff[256];
char *input = buff;
float nms=.45;
int i=0;
while(1){
if(filename){
strncpy(input, filename, 256);
image im = load_image_color(input,0,0);
image sized = letterbox_image(im, net->w, net->h);
//image sized = resize_image(im, net->w, net->h);
//image sized2 = resize_max(im, net->w);
//image sized = crop_image(sized2, -((net->w - sized2.w)/2), -((net->h - sized2.h)/2), net->w, net->h);
//resize_network(net, sized.w, sized.h);
layer l = net->layers[net->n-1];
float *X = sized.data;
time=what_time_is_it_now();
network_predict(net, X);
printf("%s: Predicted in %f seconds.
", input, what_time_is_it_now()-time);
int nboxes = 0;
detection *dets = get_network_boxes(net, im.w, im.h, thresh, hier_thresh, 0, 1, &nboxes);
//printf("%d
", nboxes);
//if (nms) do_nms_obj(boxes, probs, l.w*l.h*l.n, l.classes, nms);
if (nms) do_nms_sort(dets, nboxes, l.classes, nms);
draw_detections(im, dets, nboxes, thresh, names, alphabet, l.classes);
free_detections(dets, nboxes);
if(outfile)//
{
save_image(im, outfile);
}
else{
save_image(im, "predictions");
#ifdef OPENCV
cvNamedWindow("predictions", CV_WINDOW_NORMAL);
if(fullscreen){
cvSetWindowProperty("predictions", CV_WND_PROP_FULLSCREEN, CV_WINDOW_FULLSCREEN);
}
show_image(im, "predictions");
cvWaitKey(0);
cvDestroyAllWindows();
#endif
}
free_image(im);
free_image(sized);
if (filename) break;
}
else {
printf("Enter Image Path: ");
fflush(stdout);
input = fgets(input, 256, stdin);
if(!input) return;
strtok(input, "
");
list *plist = get_paths(input);
char **paths = (char **)list_to_array(plist);
printf("Start Testing!
");
int m = plist->size;
if(access("/home/yourname/darknet/data/out",0)==-1)//"/home/yourname/darknet/data"
{
if (mkdir("/home/yourname/darknet/data/out",0777))//"/home/yourname/darknet/data"
{
printf("creat file bag failed!!!");
}
}
for(i = 0; i < m; ++i){
char *path = paths[i];
image im = load_image_color(path,0,0);
image sized = letterbox_image(im, net->w, net->h);
//image sized = resize_image(im, net->w, net->h);
//image sized2 = resize_max(im, net->w);
//image sized = crop_image(sized2, -((net->w - sized2.w)/2), -((net->h - sized2.h)/2), net->w, net->h);
//resize_network(net, sized.w, sized.h);
layer l = net->layers[net->n-1];
float *X = sized.data;
time=what_time_is_it_now();
network_predict(net, X);
printf("Try Very Hard:");
printf("%s: Predicted in %f seconds.
", path, what_time_is_it_now()-time);
int nboxes = 0;
detection *dets = get_network_boxes(net, im.w, im.h, thresh, hier_thresh, 0, 1, &nboxes);
//printf("%d
", nboxes);
//if (nms) do_nms_obj(boxes, probs, l.w*l.h*l.n, l.classes, nms);
if (nms) do_nms_sort(dets, nboxes, l.classes, nms);
free_detections(dets, nboxes);
printf("person_num=%d
",person_num);
if(outfile){
save_image(im, outfile);
}
else{
char b[2048];
strcpy(fn,(p=strrchr(path,'/')) ? p+1 : path);
sprintf(b,"/home/yourname/darknet/data/out/%s",fn);//"/home/yourname/darknet/data"
save_image(im, b);
printf("save %s successfully!
",fn);
#ifdef OPENCV
//cvNamedWindow("predictions", CV_WINDOW_NORMAL);
//if(fullscreen){
//cvSetWindowProperty("predictions", CV_WND_PROP_FULLSCREEN, CV_WINDOW_FULLSCREEN);
//}
//show_image(im, "predictions");
// cvWaitKey(0);
// cvDestroyAllWindows();
#endif
}
free_image(im);
free_image(sized);
if (filename) break;
}
}
}
}
実行./darknet detector test cfg/voc.data cfg/yolov3-voc.cfg backup/yolov3-voc_final.Weights後に一括画像のテストを実現