python查看coco格式的box

最新推荐文章于 2024-01-23 01:05:42 发布

原创最新推荐文章于 2024-01-23 01:05:42 发布 · 205 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

小功能专栏收录该内容

22 篇文章

订阅专栏

该脚本读取指定路径下的txt和jpg文件，用OpenCV展示图片并在物体边界上画出矩形框，按f显示上一张，esc退出，其他键显示下一张。

python show_box.py <path>
图像和txt放在path目录下
按f上一张，按esc退出，其他键下一张

import os, sys, glob, cv2

path=sys.argv[1]
al = glob.glob(path+'/*txt')
al.sort()
n=len(al)
i = 0
while(1):
	a=al[i]
	b=a.replace('txt', 'jpg')
	print(i, b)
	im=cv2.imread(b)
	ih,iw,_=im.shape
	for line in open(a):
		sp=line.split()
		c=sp[0]
		x=float(sp[1])*iw
		y=float(sp[2])*ih
		w=float(sp[3])*iw
		h=float(sp[4])*ih
		x1=int(x-w/2)
		x2=int(x+w/2)
		y1=int(y-h/2)
		y2=int(y+h/2)
		cv2.rectangle(im,(x1,y1),(x2,y2),(0,0,255))
	cv2.imshow('ss', im)
	ch=cv2.waitKey() & 0xff
	if ch == 102: #f
		i -= 1
		if i < 0: i = 0
	elif ch == 27: break
	else:
		i += 1
		if i >= n: i = n-1