datatrucks解析旋转目标

女爷i 2022-09-04 02:55 176阅读 0赞

import json
    import cv2
    import numpy as np
    import math
    
    def rad(x):
      return x * np.pi / 180
    
    def cos_dist(a, b):
        if len(a) != len(b):
            return None
        part_up = 0.0
        a_sq = 0.0
        b_sq = 0.0
        for a1, b1 in zip(a, b):
            part_up += a1*b1
            a_sq += a1**2
            b_sq += b1**2
        part_down = math.sqrt(a_sq*b_sq)
        if part_down == 0.0:
            return None
        else:
            return part_up / part_down
    
    
    def order_points_quadrangle(pts):
        # sort the points based on their x-coordinates
        xSorted = pts[np.argsort(pts[:, 0]), :]
        # grab the left-most and right-most points from the sorted
        # x-roodinate points
        leftMost = xSorted[:2, :]
        rightMost = xSorted[2:, :]
        # now, sort the left-most coordinates according to their
        # y-coordinates so we can grab the top-left and bottom-left
        # points, respectively
        leftMost = leftMost[np.argsort(leftMost[:, 1]), :]
        (tl, bl) = leftMost
        # now that we have the top-left and bottom-left coordinate, use it as an
        # base vector to calculate the angles between the other two vectors
        vector_0 = np.array(bl - tl)
        vector_1 = np.array(rightMost[0] - tl)
        vector_2 = np.array(rightMost[1] - tl)
    
        angle = [np.arccos(cos_dist(vector_0, vector_1)), np.arccos(cos_dist(vector_0, vector_2))]
        (br, tr) = rightMost[np.argsort(angle), :]
        # return the coordinates in top-left, top-right,
        # bottom-right, and bottom-left order
        return np.array([tl, tr, br, bl], dtype="float32")
    
    def func(cnt):
        rect = cv2.minAreaRect(cnt) # 得到最小外接矩形的（中心(x,y), (宽,高), 旋转角度）
        box = cv2.boxPoints(rect) # 获取最小外接矩形的4个顶点坐标(ps: cv2.boxPoints(rect) for OpenCV 3.x)
        box = np.int0(box)
        return box
    
    with open('/home/lhq/segmentation/snake-master/icecream_6_8.json') as f:
        datas = f.readlines()
        for data in datas:
            data = data.strip()
            Dict = json.loads(data)
    
            content = Dict['content'].split('___dataturks_')[-1]
            Dict['content'] = content
            fw = open('patchs/{}.txt'.format(content.split('.')[0]),'w')
            annotations = Dict['annotation']
            for annotation in annotations:
                imageWidth = annotation['imageWidth']
                imageHeight = annotation['imageHeight']
                points = annotation['points']
                for point in points:
                    point[0] = int(point[0] * imageWidth)
                    point[1] = int(point[1] * imageHeight)
                bbox = func(np.array(points))
                try:
                    bbox = order_points_quadrangle(bbox)
                    fw.write('0 ')
                    for box in bbox:
                        fw.write(str(float(box[0] / imageWidth)) + ' ' + str(float(box[1] / imageHeight)) + ' ')
                    fw.write('\n')
                except:
                    print(content)