图片旋转90,180,270

import cv2
import numpy as np
import os
def rotation(img,i,save_path,angle):
    width = img.shape[0]  # 长度
    height = img.shape[1]  # 宽度
    # 这里的第一个参数为旋转中心,第二个为旋转角度,第三个为旋转后的缩放因子
    # 可以通过设置旋转中心,缩放因子,以及窗口大小来防止旋转后超出边界的问题
    M = cv2.getRotationMatrix2D((width/2, height/2), angle, 1)
    # 第三个参数是输出图像的尺寸中心
    dst = cv2.warpAffine(img, M, (width, height))
    img_rotate_name = str(angle)+'_'+i
    img_rotate_path = os.path.join(save_path,img_rotate_name)
    cv2.imwrite(img_rotate_path, dst)
if __name__ =="__main__":
    paths = ('F:/AI/glaucoma_set/gla_img_120_Chall/Annotation-Training400/Disc_Cup_Fovea_Illustration_512/',
          'F:/AI/glaucoma_set/gla_img_120_Chall/Annotation-Training400/Disc_Cup_Masks/Glaucoma_512_012/',
          'F:/AI/glaucoma_set/gla_img_120_Chall/Annotation-Training400/Disc_Cup_Masks/Non-Glaucoma_512_012')

    # save_paths = ('F:/AI/glaucoma_set/gla_img_120_Chall/Annotation-Training400/Disc_Cup_Fovea_Illustration_512/rotate/',
    #       'F:/AI/glaucoma_set/gla_img_120_Chall/Annotation-Training400/Disc_Cup_Masks/Glaucoma_512_012/rotate/',
    #       'F:/AI/glaucoma_set/gla_img_120_Chall/Annotation-Training400/Disc_Cup_Masks/Non-Glaucoma_512_012/rotate')

    for path in paths:
        save_path = path+'rotate'
        name = os.listdir(path)
        for i in name:
            rotation_angle = [90, 180, 270]
            for angle in rotation_angle:
                image_path = os.path.join(path, i)
                image = cv2.imread(image_path, cv2.IMREAD_ANYCOLOR)
                img = rotation(image,i,save_path,angle)
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值