diff --git a/sort.py b/sort.py index 4b2ca8a..d87d58a 100644 --- a/sort.py +++ b/sort.py @@ -317,6 +317,7 @@ def parse_args(): print('%d,%d,%.2f,%.2f,%.2f,%.2f,1,-1,-1,-1'%(frame,d[4],d[0],d[1],d[2]-d[0],d[3]-d[1]),file=out_file) if(display): d = d.astype(np.int32) + ax1.text(d[0],d[1],str(d[4]),c="white",bbox=dict(facecolor="green")) ax1.add_patch(patches.Rectangle((d[0],d[1]),d[2]-d[0],d[3]-d[1],fill=False,lw=3,ec=colours[d[4]%32,:])) if(display):