Browse Source

修改ai.py

master
chenyukun 1 year ago
parent
commit
0aec966fb3
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      AI.py

+ 1
- 1
AI.py View File

x1=np.max( bpoints[:,0] ) x1=np.max( bpoints[:,0] )
y1=np.max( bpoints[:,1] ) y1=np.max( bpoints[:,1] )
conf= ((x0+x1)/W + (y0+y1)/H)/4.0; conf= ((x0+x1)/W + (y0+y1)/H)/4.0;
conf=math.fabs((conf-0.5)/0.5)
conf=1.0 - math.fabs((conf-0.5)/0.5)
xyxy=[x0,y0,x1,y1] xyxy=[x0,y0,x1,y1]
xyxy=[int(x+0.5) for x in xyxy] xyxy=[int(x+0.5) for x in xyxy]
#float(cls_c), *xywh, float(conf_c)] #float(cls_c), *xywh, float(conf_c)]

Loading…
Cancel
Save