11、Python對Labeme數據集進行擴充----圖片旋轉及座標對應映射

問題描述：存在一組labelme標註的圖片和json串，然後進行旋轉產生新圖片和相應的json文件；

# -*- coding: utf-8 -*-
import os
import sys
import json
import io
import random
import re
import cv2
import numpy as np
from random import choice
import math
 
source_path='./mac'
destination_path='./rotate'
angle = [90,180,270] 
 
article_info = {}
data_json = json.loads(json.dumps(article_info))
data_json['version']='3.6.16'
data_json['flags']={}
 
data_json["lineColor"]=[
    0,
    255,
    0,
    128
  ]
data_json["fillColor"]=[
    255,
    0,
    0,
    128
  ]
 
 
 
def file_name(file_dir):
    L = []
    for root, dirs, files in os.walk(file_dir):
        for file in files:
            if os.path.splitext(file)[1] == '.json':
                L.append(os.path.join(root , file))
        return L
 
 
 
def rotation_point(img,angle,point):
        cols = img.shape[1]
        rows = img.shape[0]
        M = cv2.getRotationMatrix2D((cols / 2, rows / 2), angle, 1)
        heightNew = int(cols * math.fabs(math.sin(math.radians(angle))) + rows* math.fabs(math.cos(math.radians(angle))))
        widthNew = int(rows * math.fabs(math.sin(math.radians(angle))) + cols* math.fabs(math.cos(math.radians(angle))))
        M[0, 2] += (widthNew - cols) / 2  
        M[1, 2] += (heightNew - rows) / 2 
        img = cv2.warpAffine(img, M, (widthNew, heightNew))
        a = M[:, :2] 
        b = M[:, 2:]
        b = np.reshape(b, newshape=(1, 2))
        a = np.transpose(a)
        point = np.reshape(point, newshape=(int(len(point) /2), 2))
        point = np.dot(point, a) + b
        point = np.reshape(point, newshape=(int(len(point) / 4), 8))
        return img, point
 
 
for name in enumerate(file_name(source_path)): 
    shape_json = []
    m_path =name[1]
    dir=os.path.dirname(m_path)
    file_json = io.open(m_path,'r',encoding='utf-8')
    json_data = file_json.read()
    data = json.loads(json_data)
    data_json['imageData']=None
    data_name= data['imagePath']
    data_path = dir+'/'+data_name
    angle_item=choice(angle)
    object_name= os.path.splitext(data['imagePath'])[0]
    for i in  range(len(data['shapes'])):
        m_xmin_0=(data['shapes'][i]['points'][0][0]  if(data['shapes'][i]['points'][0][0]<data['shapes'][i]['points'][1][0]) else data['shapes'][i]['points'][1][0])
        print ('m_xmin_0=',m_xmin_0)
        m_ymin_0=(data['shapes'][i]['points'][0][1]  if(data['shapes'][i]['points'][0][1]<data['shapes'][i]['points'][1][1]) else data['shapes'][i]['points'][1][1])
        print ('m_ymin_0=',m_ymin_0)
        m_xmax_0=(data['shapes'][i]['points'][1][0]  if(data['shapes'][i]['points'][0][0]<data['shapes'][i]['points'][1][0]) else data['shapes'][i]['points'][0][0])
        print ('m_xmax_0=',m_xmax_0)
        m_ymax_0=(data['shapes'][i]['points'][1][1]  if(data['shapes'][i]['points'][0][1]<data['shapes'][i]['points'][1][1]) else data['shapes'][i]['points'][0][1])
        print ('m_ymax_0=',m_ymax_0 )        
        m_name_0=data['shapes'][i]['label']
        print ('m_name_0=',m_name_0)
        data_json_line_color=data['shapes'][i]['line_color']
        data_json_fill_color=data['shapes'][i]['fill_color']
        data_json_rec=data['shapes'][i]['shape_type']
        img=cv2.imread(data_path)
        point=np.array([m_xmin_0,m_ymin_0,m_xmin_0,m_ymax_0,m_xmax_0,m_ymin_0,m_xmax_0,m_ymax_0])
        im_rotate,point = rotation_point(img,angle_item,point )
        print (point)
        (filename, extension) = os.path.splitext(data_name)
        data_new_picture_name=destination_path+"/"+filename+"_rotate"+".jpg"
        data_new_json_name =destination_path+"/"+filename+"_rotate"+".json"
        data_json['imagePath']=filename+"_rotate"+".jpg"
        cv2.imwrite(data_new_picture_name,im_rotate)
        im_rotate=cv2.imread(data_new_picture_name) 
        data_json['imageWidth']=im_rotate.shape[1]
        data_json['imageHeight']=im_rotate.shape[0]
        data_json_min_x=point[0][0]
        data_json_min_y=point[0][1]
        data_json_max_x=point[0][-2]
        data_json_max_y=point[0][-1]
        shape_json_item={"label": m_name_0,"line_color": data_json_line_color,"fill_color": data_json_fill_color,"points": [[data_json_min_x,data_json_min_y], 
        [data_json_max_x,data_json_max_y]],"shape_type": data_json_rec}
        shape_json.append(shape_json_item)
    data_json['shapes']=shape_json
    data_info = json.dumps(data_json, ensure_ascii=False)
    fp=open(data_new_json_name, "w+")
    json.dump(data_info,fp,ensure_ascii=False,indent =4)
    fp.close()
    fp=open(data_new_json_name, "r")  
    for x in fp.readlines():
        y=x.replace("\\\"","\"")
        z=y.replace("\"{","{")
        w=z.replace("}\"","}")  
        fp.close()
        fp=open(data_new_json_name, "w+")
        fp.write(w)
        fp.close()