查看四边形四个点的顺序是否正确

最新推荐文章于 2022-05-09 21:19:24 发布

Alphapeople

最新推荐文章于 2022-05-09 21:19:24 发布

阅读量686

点赞数

分类专栏： python

本文链接：https://blog.csdn.net/weixin_38241876/article/details/112555483

版权

python 专栏收录该内容

112 篇文章 1 订阅

订阅专栏

imagespaths = []
boxpaths = []
basedir = '/home/lixuan/centernettrain/train_data/txt标注样列'
for dir in os.listdir(basedir):
    if 'jpg' in dir:
        imagespaths.append(os.path.join(basedir, dir))
    else:
        boxpaths.append(os.path.join(basedir, dir))

for path in imagespaths:
    txtpath = path.replace('jpg', 'txt')
    with open(txtpath)  as f:
        data = f.read().split(' ')
        boxs = [data[i:i + 8] for i in range(0, len(data), 8)][:-1]
    img = cv2.imread(path)
    for box in boxs:  # cv2.circle(image, (int(point[0]), int(point[1])), 10, (255, 0, 0), -1)
        box = [int(float(x)) for x in box]
        box = [box[i:i + 2] for i in range(0, len(box), 2)]
        img = cv2.circle(img, (int(box[0][0]), int(box[0][1])), 10, (255, 0, 0), -1)
        img = cv2.circle(img, (int(box[1][0]), int(box[1][1])), 10, (0, 255, 0), -1)
        img = cv2.circle(img, (int(box[2][0]), int(box[2][1])), 10, (0, 0, 255), -1)
        img = cv2.circle(img, (int(box[3][0]), int(box[3][1])), 10, (255, 255, 0), -1)
    cv2.imwrite(path.replace(basedir, '/home/lixuan/桌面/multicenternetcopy/look'), img)
    # # img = cv2.resize(img,(w//4,h//4))
    # # cv2.imshow('',img)
    # # cv2.waitKey(0)