IOS swift 原始横向图片旋转为竖向旋转

2020-05-08 04:16:35   计算机视觉
  ios   Swift  

iOS相机默认输出是个横向图,需要将其转为竖向,网上试了一些方法效率都不是很高,因为在做实时的图像算法,所以对FPS要求很高,用accelerate框架自己写了一个转竖向图的方法

基于CVPixelBuffer写了一个extension

extension CVPixelBuffer {
    public func ro() -> CVPixelBuffer? {
        let size = CGRect(origin: CGPoint(x: 0, y: 0), size: self.size)

        let inputImageRowBytes = CVPixelBufferGetBytesPerRow(self)
        let imageChannels = 4

        CVPixelBufferLockBaseAddress(self, CVPixelBufferLockFlags(rawValue: 0))
        defer { CVPixelBufferUnlockBaseAddress(self, CVPixelBufferLockFlags(rawValue: 0)) }

        // Finds the address of the upper leftmost pixel of the source area.
        guard
          let inputBaseAddress = CVPixelBufferGetBaseAddress(self)?.advanced(
            by: Int(size.minY) * inputImageRowBytes + Int(size.minX) * imageChannels)
        else {
          return nil
        }

        // Crops given area as vImage Buffer.
        var croppedImage = vImage_Buffer(
          data: inputBaseAddress, height: UInt(self.size.height), width: UInt(self.size.width),
          rowBytes: inputImageRowBytes)

        // ======

        let paddedRotatedRowBytes = Int(size.height) * imageChannels
        guard let paddedRotatedAddress = malloc(Int(size.width) * paddedRotatedRowBytes) else {
          return nil
        }

        // Allocates a vacant vImage buffer for resized image.
        var paddedRotatedImage = vImage_Buffer(
          data: paddedRotatedAddress,
          height: UInt(size.width), width: UInt(size.height),
          rowBytes: paddedRotatedRowBytes
        )

        var cg_transform = vImage_CGAffineTransform(a: 0.0, b: -1, c: 1, d: 0.0, tx: Double(0), ty: Double(size.width))
        guard vImageAffineWarpCG_ARGB8888(&croppedImage, &paddedRotatedImage, nil, &cg_transform, clear_color, vImage_Flags(0)) == kvImageNoError else {
            free(paddedRotatedAddress)
            return nil
        }

        let releaseCallBack: CVPixelBufferReleaseBytesCallback = { mutablePointer, pointer in
          if let pointer = pointer {
            free(UnsafeMutableRawPointer(mutating: pointer))
          }
        }

        var result: CVPixelBuffer?

        // Converts the thumbnail vImage buffer to CVPixelBuffer
        let conversionStatus = CVPixelBufferCreateWithBytes(
          nil,
          Int(size.height), Int(size.width),
          CVPixelBufferGetPixelFormatType(self),
          paddedRotatedAddress,
          paddedRotatedRowBytes,
          releaseCallBack,
          nil,
          nil,
          &result
        )

        guard conversionStatus == kCVReturnSuccess else {
          free(paddedRotatedAddress)
          return nil
        }

        return result
      }
}

调用方法也很简单,只需要直接.ro即可

guard let rPixelBuffer = oPixelBuffer.ro() else {
        return nil
}
viencoding.com版权所有,允许转载,但转载请注明出处和原文链接: https://viencoding.com/article/266
欢迎小伙伴们在下方评论区留言 ~ O(∩_∩)O
文章对我有帮助, 点此请博主吃包辣条 ~ O(∩_∩)O

猜你喜欢


评论

There are no comments yet.
未登录

登录后即可发表评论

登录或注册

标签

AdSense Anaconda Android API apt Auth AWS B-tree Bandwagon Blog bower brew bytes Caffe Catalina cloudcone Composer conda CPU crontab CSS Cuda cv2 datetime Digitalocean DNS Docker Docker-Compose export Flask FTP GET Git GitHub Gmail GoDaddy Google GTM hash Homebrew Homestead HTML http HTTPS IDEA image imagemagick imagick imgick import InnoDB ios iPhone ISO8601 iTerm2 Java JPG Keras Laravel Laravel-Admin lazyload Linux list lnmp load logs Lravel Mac Markdown matplotlib md5 mix MobileNet Mojave mongo MongoDB MySQL Namesilo Nginx npm numpy Nvidia Nvidia-Docker onevps OpenCV Openpose Outline parse PayPal PHP php-fpm PhpStorm PHP扩展 PIL Pillow pip PNG POST Protobuf PyCharm pyenv pymongo Python Python,人工智能,机器学习,VOC,xml Queue Redis requests RGB save selenium SEO Shadowsock Shadowsocks ShadowsocksR simplemde Spring Boot SQLServer ssd SSH SSL证书 SSR str Sublime sudo swap Swift Tensorflow Terminal Terminator Ubuntu urllib UTC v2ray Valet Validation Validator VienBlog virtualenvs VPN VPS Vultr Web Windows WordPress Xcode yaml YUV zip zmq zsh 下载图片 云主机 云服务器 人工智能 优化 优惠码 伪原创 作弊与反作弊 免费ss账号 免费提现 切片 前端 加密 协议 博客 友链 双击事件 后台运行 后端 命令 图片操作 图片转换 域名 大小写转换 姿态检测 安卓模拟器 安装 定时任务 定时执行 导出导入 小程序码 延迟加载 微信小程序 慢查询 懒加载 提现 搜索引擎 搬瓦工 搭梯子 教程 数据库 文件上传 无法登录 日志 日期 时区 时间 时间戳 服务器 机器学习 权限 梯子 模拟浏览器 港版支付宝 爬虫 生活服务 科学上网 系统升级 索引 组件开发 编辑器 自动付款 英文伪原创 计划任务 计算机视觉 语法 远程连接 配置文件 重定向 错误异常 错误提示 队列 阿里云 香港 香港手机号
亲情非友情链接