參考了這篇:https://blog.csdn.net/dao_0123/article/details/102967244
但是上面那篇把單次輸入路徑的方式給覆蓋了,
我這裏偷了個懶,就是套了一層if else,實現可以單次或者批量輸入路徑進行測試,
下面這個是參考的那篇文章:
1.請在/darknet/src路徑下找到detector.c文件,並打開,或請在根目錄下直接執行:
2.定位到 void test_detector(char *datacfg …)這一行,將這個函數模塊用以下代碼完全替代:注意3個地方:
(1)請在根路徑下新建一個測試文檔,本文新建了test.txt文件,裏面是你需要測試圖片的路徑與文件名。(2)請在根目錄下新建文件夾,用來存放測試圖片的結果,本文在根目錄下新建文件夾,命名爲result_img。讀者可自己定義。
(3)請在以下代碼 sprintf(b, “result_img/%d”, i); 處修改成對應自己的文件夾路徑。
3.更改圖片命名,在代碼中添加GetFilename(),函數
————————————————
版權聲明:本文爲CSDN博主「小晴天明」的原創文章,遵循CC 4.0 BY-SA版權協議,轉載請附上原文出處鏈接及本聲明。
原文鏈接:https://blog.csdn.net/dao_0123/article/details/102967244
如果自己不想改,可以直接複製下面的代碼,然後替換到test_detector部分,注意上面的那個更改圖片命名的是單另一個函數,放在一起方便複製。
還有一點值得注意的是,這裏的test_detector的入口,是12個參數,需要看清楚,自己下的版本有幾個參數,要不然改完就會報錯
//更改圖片命名
char *GetFilename(char *p)
{
static char name[40] = { "" };
char *q = strrchr(p, '/') + 1;
strncpy(name, q, 35);//後面的35是圖片名長度(不包括後綴),根據自己的需要進行修改
return name;
}
//
void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh,
float hier_thresh, int dont_show, int ext_output, int save_labels, char *outfile, int letter_box, int benchmark_layers)
{
list *options = read_data_cfg(datacfg);
char *name_list = option_find_str(options, "names", "data/names.list");
int names_size = 0;
char **names = get_labels_custom(name_list, &names_size); //get_labels(name_list);
image **alphabet = load_alphabet();
network net = parse_network_cfg_custom(cfgfile, 1, 1); // set batch=1
if (weightfile) {
load_weights(&net, weightfile);
}
net.benchmark_layers = benchmark_layers;
fuse_conv_batchnorm(net);
calculate_binary_weights(net);
if (net.layers[net.n - 1].classes != names_size) {
printf("\n Error: in the file %s number of names %d that isn't equal to classes=%d in the file %s \n",
name_list, names_size, net.layers[net.n - 1].classes, cfgfile);
if (net.layers[net.n - 1].classes > names_size) getchar();
}
srand(2222222);
char buff[256];
char *input = buff;
char *json_buf = NULL;
int json_image_id = 0;
FILE* json_file = NULL;
if (outfile) {
json_file = fopen(outfile, "wb");
if(!json_file) {
error("fopen failed");
}
char *tmp = "[\n";
fwrite(tmp, sizeof(char), strlen(tmp), json_file);
}
int j;
float nms = .45; // 0.4F
//開始循環//
int i;
while (1) {
if (filename) {
strncpy(input, filename, 256);
list *plist = get_paths(input);
char **paths = (char **)list_to_array(plist);
printf("Start Testing!\n");
int m = plist->size;
for (i = 0; i < m; ++i) {
char *path = paths[i];
image im = load_image(path, 0, 0, net.c);
int letterbox = 0;
image sized = resize_image(im, net.w, net.h);
//image sized = letterbox_image(im, net.w, net.h); letterbox = 1;
layer l = net.layers[net.n - 1];
float *X = sized.data;
double time = get_time_point();
network_predict(net, X);
printf("%s: Predicted in %lf milli-seconds.\n", input, ((double)get_time_point() - time) / 1000);
printf("Try Very Hard:");
printf("%s: Predicted in %lf milli-seconds.\n", path, ((double)get_time_point() - time) / 1000);
int nboxes = 0;
detection *dets = get_network_boxes(&net, im.w, im.h, thresh, hier_thresh, 0, 1, &nboxes, letterbox);
if (nms) do_nms_sort(dets, nboxes, l.classes, nms);
// draw_detections_v3(basecfg(input), im, dets, nboxes, thresh, names, alphabet, l.classes, ext_output);
draw_detections_v3(im, dets, nboxes, thresh, names, alphabet, l.classes, ext_output);
char b[2048];
sprintf(b, "result_img/%s", GetFilename(path)); //***改成自己的文件夾路徑
save_image(im, b);
printf("save %s successfully!\n", GetFilename(path));//文件命名在這個地方,可改爲i
if (save_labels)
{
char labelpath[4096];
replace_image_to_label(input, labelpath);
FILE* fw = fopen(labelpath, "wb");
int i;
for (i = 0; i < nboxes; ++i) {
char buff[1024];
int class_id = -1;
float prob = 0;
for (j = 0; j < l.classes; ++j) {
if (dets[i].prob[j] > thresh && dets[i].prob[j] > prob) {
prob = dets[i].prob[j];
class_id = j;
}
}
if (class_id >= 0) {
sprintf(buff, "%d %2.4f %2.4f %2.4f %2.4f\n", class_id, dets[i].bbox.x, dets[i].bbox.y, dets[i].bbox.w, dets[i].bbox.h);
fwrite(buff, sizeof(char), strlen(buff), fw);
}
}
fclose(fw);
}
free_detections(dets, nboxes);
free_image(im);
free_image(sized);
}
printf("All Done!\n");
exit(0);
}
else {
//這裏這個if可以刪掉,爲了更好地看到哪裏改動了,這塊我就沒有刪。
if (filename) {
strncpy(input, filename, 256);
if (strlen(input) > 0)
if (input[strlen(input) - 1] == 0x0d) input[strlen(input) - 1] = 0;
}
else {
printf("Enter Image Path: ");
fflush(stdout);
input = fgets(input, 256, stdin);
if (!input) break;
strtok(input, "\n");
}
//image im;
//image sized = load_image_resize(input, net.w, net.h, net.c, &im);
image im = load_image(input, 0, 0, net.c);
image sized;
if (letter_box) sized = letterbox_image(im, net.w, net.h);
else sized = resize_image(im, net.w, net.h);
layer l = net.layers[net.n - 1];
//box *boxes = calloc(l.w*l.h*l.n, sizeof(box));
//float **probs = calloc(l.w*l.h*l.n, sizeof(float*));
//for(j = 0; j < l.w*l.h*l.n; ++j) probs[j] = (float*)xcalloc(l.classes, sizeof(float));
float *X = sized.data;
//time= what_time_is_it_now();
double time = get_time_point();
network_predict(net, X);
//network_predict_image(&net, im); letterbox = 1;
printf("%s: Predicted in %lf milli-seconds.\n", input, ((double)get_time_point() - time) / 1000);
//printf("%s: Predicted in %f seconds.\n", 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, letter_box);
if (nms) {
if (l.nms_kind == DEFAULT_NMS) do_nms_sort(dets, nboxes, l.classes, nms);
else diounms_sort(dets, nboxes, l.classes, nms, l.nms_kind, l.beta_nms);
}
draw_detections_v3(im, dets, nboxes, thresh, names, alphabet, l.classes, ext_output);
save_image(im, "predictions");
if (!dont_show) {
show_image(im, "predictions");
}
if (json_file) {
if (json_buf) {
char *tmp = ", \n";
fwrite(tmp, sizeof(char), strlen(tmp), json_file);
}
++json_image_id;
json_buf = detection_to_json(dets, nboxes, l.classes, names, json_image_id, input);
fwrite(json_buf, sizeof(char), strlen(json_buf), json_file);
free(json_buf);
}
// pseudo labeling concept - fast.ai
if (save_labels)
{
char labelpath[4096];
replace_image_to_label(input, labelpath);
FILE* fw = fopen(labelpath, "wb");
int i;
for (i = 0; i < nboxes; ++i) {
char buff[1024];
int class_id = -1;
float prob = 0;
for (j = 0; j < l.classes; ++j) {
if (dets[i].prob[j] > thresh && dets[i].prob[j] > prob) {
prob = dets[i].prob[j];
class_id = j;
}
}
if (class_id >= 0) {
sprintf(buff, "%d %2.4f %2.4f %2.4f %2.4f\n", class_id, dets[i].bbox.x, dets[i].bbox.y, dets[i].bbox.w, dets[i].bbox.h);
fwrite(buff, sizeof(char), strlen(buff), fw);
}
}
fclose(fw);
}
free_detections(dets, nboxes);
free_image(im);
free_image(sized);
if (!dont_show) {
wait_until_press_key_cv();
destroy_all_windows_cv();
}
if (filename) break;
}
}
//這裏之前的是循環部分//
if (json_file) {
char *tmp = "\n]";
fwrite(tmp, sizeof(char), strlen(tmp), json_file);
fclose(json_file);
}
// free memory
free_ptrs((void**)names, net.layers[net.n - 1].classes);
free_list_contents_kvp(options);
free_list(options);
const int nsize = 8;
for (j = 0; j < nsize; ++j) {
for (i = 32; i < 127; ++i) {
free_image(alphabet[j][i]);
}
free(alphabet[j]);
}
free(alphabet);
free_network(net);
}
完成之後,運行:
注意那些data、cfg、weights的路徑都是自己的
還有這個test.txt要放在運行darknet的下面,或者寫test的絕對路徑
darknet.exe detector test cfg/xxx.data yolov3.cfg backup/yolov3-groundbox_final.weights test.txt
只能測單張的問題:
這裏有個問題需要注意一下,在做test.txt的時候,需要是反斜槓/,也就是在ubuntu的寫法,我在win上自己生成測試list的時候,因爲用了\\,造成只能測試一張圖片的問題,然後我還以爲是名字必須是ImageSets,改了之後發現不是
import os
class ReName:
def __init__(self,filePath,startNum):
self.filePath=filePath
self.startNum=startNum
def RName(self):
fileNames = os.listdir(self.filePath)
i=0
for name in fileNames:
if name[-3:]=='jpg':
print('change')
newName=str("%06d" %(self.startNum+i))#"%06d" % int(id)
os.rename(name,newName+'.jpg')
i+=1
return 'finish'
def list_make(self):
filePath=os.listdir(self.filePath)
print(self.filePath)
with open('testlist.txt','w')as f:
for file in filePath:
f.write(self.filePath+file+'\n')
f.close()
return 'listmake'
if __name__=='__main__':
startNum=58 #這裏根據自己的需要,改名字的起始數字
filePath='./text/VOCdevkit/VOC2018/JPEGImages/'#改成自己的,注意用反斜槓
re=ReName(filePath,startNum)
# print(re.RName())
print(re.list_make())