1.代码
#!/usr/bin/python3
import cv2 as cv
import os
# 图片文件夹
dir_path = "/home/newj/下载/Basketball/img/"
# 目标标注框,ground truth
ground_truth_path = "/home/newj/下载/Basketball/groundtruth_rect.txt"
# 将文件夹下的所有文件名保存到一个list
imgs_list = os.listdir(dir_path)
# 因为listdir保存到列表,顺序被打乱了
# 按照文件名排序
imgs_list.sort()
# print(np.size(imgs_list))
# 新建文件对象,打开保存有ground truth的txx
f = open(ground_truth_path)
# 按行读取文件,保存为list,list每一个元素为str,之后要进行格式转换
# 其中的一个元素'117,213,34,81\n'
lines = f.readlines()
# print(np.size(lines))
for name, rect in zip(imgs_list, lines):
# 获取图片路径
img_path = os.path.join(dir_path, name)
print(img_path)
# 读取图片
img = cv.imread(img_path)
# 去掉元素末尾的'\n'
rect = rect.strip('\n')
# 按','分割字符串为列表,其中每一个元素为字符串
rect = rect.split(',')
print(rect)
# 画框
cv.rectangle(img, (int(rect[0]), int(rect[1])), (int(rect[0]) + int(rect[2]), (int(rect[1]) + int(rect[3]))),
(0, 0, 255), 2)
# 显示
cv.imshow('Basketball_Sequence', img)
# 阻塞等待
# 按下任意键,播放下一帧
cv.waitKey(0)
2.效果
basketball play