前提・実現したいこと
YOLOv3を使って自作学習データで動画の検出はできたがそれらのBBOXの座標をtxtに出力させるのに詰まっている
該当のソースコード
void draw_detections(image im, detection dets, int num, float thresh, char **names, image **alphabet, int classes)
{
int i,j,q=-1;
for(i = 0; i < num; ++i){
char labelstr[4096] = {0};
int class = -1;
for(j = 0; j < classes; ++j){
if (dets[i].prob[j] > thresh){
if (class < 0) {
strcat(labelstr, names[j]);
class = j;
} else {
strcat(labelstr, ", ");
strcat(labelstr, names[j]);
}
q+=1;
int count[q][4];
printf("%s: %.0f%%\n", names[j], dets[i].prob[j]100);
printf(" -> (x, y, w, h)\n");
//x=(int)(dets[i].bbox.xim.w);
count[q][0]=(int)(dets[i].bbox.xim.w);
count[q][1]=(int)(dets[i].bbox.yim.h);
count[q][2]=(int)(dets[i].bbox.wim.w);
count[q][3]=(int)(dets[i].bbox.h*im.h);
printf("x=%d, y=%d, w=%d,h=%d\n",count[q][0],count[q][1],count[q][2],count[q][3]);\ここをtxtに出力
} } if(class >= 0){ int width = im.h * .006; /* if(0){ width = pow(prob, 1./2.)*10+1; alphabet = 0; } */ //printf("%d %s: %.0f%%\n", i, names[class], prob*100); int offset = class*123457 % classes; float red = get_color(2,offset,classes); float green = get_color(1,offset,classes); float blue = get_color(0,offset,classes); float rgb[3]; //width = prob*20+2; rgb[0] = red; rgb[1] = green; rgb[2] = blue; box b = dets[i].bbox; //printf("%f %f %f %f\n", b.x, b.y, b.w, b.h); int left = (b.x-b.w/2.)*im.w; int right = (b.x+b.w/2.)*im.w; int top = (b.y-b.h/2.)*im.h; int bot = (b.y+b.h/2.)*im.h; if(left < 0) left = 0; if(right > im.w-1) right = im.w-1; if(top < 0) top = 0; if(bot > im.h-1) bot = im.h-1;
試したこと
結構無理矢理な書き方ですけど一応このコードだとコマンドラインには座標が表示されていっている。
↓いまコマンドラインに表示されている結果。これらをtxtファイルに投げたい
FPS:34.1
Objects:
KPHN: 93%
-> (x, y, w, h)
x=799, y=942, w=100,h=40
FPS:35.7
Objects:
KPHN: 94%
-> (x, y, w, h)
x=798, y=941, w=98,h=40
回答1件
あなたの回答
tips
プレビュー