会使用到PIL,os,pandas库
from PIL import ImageDraw
from PIL import Image
import os
import pandas as pd
import numpy as np
我们首先需要处理txt文件,我的txt文件为边界框的四个点坐标,例如(0,0,1.0,2.69)。
def read_bb_anno(seq_path):
bb_anno_file = os.path.join(seq_path)
gt = pd.read_csv(bb_anno_file, delimiter=',', usecols=[0,1,2,3], header=None, na_filter=False, low_memory=False)
col = list(gt.columns)
gt[col] = gt[col].apply(pd.to_numeric, errors='coerce').fillna(0)
gt = np.array(gt, dtype=np.float32)
for item in gt:
item[2] += item[0]
item[3] += item[1]
return gt
我的任务需要处理两个txt文件,看大家具体任务需求。通过获取到的坐标来绘制边界框
class drawRect():
def __init__(self):
self.groundtruth = ''
self.imagePath = ''
self.resultPat