码迷,mamicode.com
首页 > 其他好文 > 详细

旋转图片,空余部分补黑色,并将原坐标转换

时间:2018-10-11 11:31:51      阅读:202      评论:0      收藏:0      [点我收藏+]

标签:计算   角度   .com   port   draw   坐标转换   for   oid   http   

旋转图片,空余部分补黑色,并将原坐标转换

#载入必要的模块
import os
from PIL import Image
from PIL import ImageDraw
import sys
import xlrd
import numpy
import math
import xlsxwriter

#输入
ExcelFile=xlrd.open_workbook(r.\标记信息.xlsx)
sheet=ExcelFile.sheet_by_index(0)
#输出
workbook = xlsxwriter.Workbook(./new/旋转后的标记信息.xlsx) #创建工作簿
worksheet = workbook.add_worksheet() #创建工作表



print(sheet.nrows)

for i in range(0,sheet.nrows):
    videoid=str(int(sheet.cell_value(i,0)))
    frameid=str(int(sheet.cell_value(i,1)))
    x1=int(sheet.cell_value(i,2))
    x2=int(sheet.cell_value(i,4))
    x3=int(sheet.cell_value(i,6))
    x4=int(sheet.cell_value(i,8))
    y1=int(sheet.cell_value(i,3))
    y2=int(sheet.cell_value(i,5))
    y3=int(sheet.cell_value(i,7))
    y4=int(sheet.cell_value(i,9))
    bookname=sheet.cell_value(i,10)

    #print([x1,y1,x2,y2,x4,y4,x3,y3,x1,y1])


    pic_fole_head =  Image.open(videoid+"_"+frameid+".jpg")
    width,height=pic_fole_head.size
    #print([width,height])

    #计算余弦值
    cosa=(y3-y1)/numpy.sqrt(numpy.power(x1-x3,2)+numpy.power(y1-y3,2))
    #计算需要选装的弧度
    rad=numpy.abs(numpy.arccos(cosa))
    #计算需要旋转的角度
    alpha=rad*180/math.pi
    #判断旋转的方向
    #并且计算旋转后的坐标
    if x3<x1:
        pic_fole_head=pic_fole_head.rotate(alpha,0,1)
        yy1=y1*numpy.cos(rad)+(width-x1)*numpy.sin(rad)
        xx1=x1*numpy.cos(rad)+y1*numpy.sin(rad)
        yy2=y2*numpy.cos(rad)+(width-x2)*numpy.sin(rad)
        xx2=x2*numpy.cos(rad)+y1*numpy.sin(rad)
        yy3=y3*numpy.cos(rad)+(width-x3)*numpy.sin(rad)
        xx3=x3*numpy.cos(rad)+y3*numpy.sin(rad)
        yy4=y4*numpy.cos(rad)+(width-x4)*numpy.sin(rad)
        xx4=x4*numpy.cos(rad)+y4*numpy.sin(rad)
    else:
        alpha=-alpha
        pic_fole_head=pic_fole_head.rotate(alpha,0,1)
        yy1=y1*numpy.cos(rad)+x1*numpy.sin(rad)
        xx1=x1*numpy.cos(rad)+(height-y1)*numpy.sin(rad)
        yy2=y2*numpy.cos(rad)+x2*numpy.sin(rad)
        xx2=x2*numpy.cos(rad)+(height-y2)*numpy.sin(rad)
        yy3=y3*numpy.cos(rad)+x3*numpy.sin(rad)
        xx3=x3*numpy.cos(rad)+(height-y3)*numpy.sin(rad)
        yy4=y4*numpy.cos(rad)+x4*numpy.sin(rad)
        xx4=x4*numpy.cos(rad)+(height-y4)*numpy.sin(rad)

    #print([xx1,yy1,xx2,yy2,xx4,yy4,xx3,yy3,xx1,yy1])
    #框出旋转后的书名,测试时用来验证
    #drawObject=ImageDraw.Draw(pic_fole_head)
    #drawObject.line([xx1,yy1,xx2,yy2,xx4,yy4,xx3,yy3,xx1,yy1],(0,5,255),5)
    
    #保存新结果
    worksheet.write(i, 0, int(videoid))
    worksheet.write(i, 1, int(frameid))
    worksheet.write(i, 2, int(xx1))
    worksheet.write(i, 3, int(yy1))
    worksheet.write(i, 4, int(xx2))
    worksheet.write(i, 5, int(yy2))
    worksheet.write(i, 6, int(xx3))
    worksheet.write(i, 7, int(yy3))
    worksheet.write(i, 8, int(xx4))
    worksheet.write(i, 9, int(yy4))
    worksheet.write(i, 10, bookname)
    worksheet.write(i, 11, videoid+"_"+frameid+"_"+str(i+1)+".png")
    #pic_fole_head.show()

    pic_fole_head.save("./new/"+videoid+"_"+frameid+"_"+str(i+1)+".jpg")
workbook.close()

本程序目的是:将标记的书名放到垂直方向

载入xlsx的部分因人而异,可以只看循环体中转换的步骤,坐标的转换是一个简单的几何关系。

最终的效果如下:

技术分享图片》》》》技术分享图片

 

旋转图片,空余部分补黑色,并将原坐标转换

标签:计算   角度   .com   port   draw   坐标转换   for   oid   http   

原文地址:https://www.cnblogs.com/littleby/p/9770949.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!