博客
关于我
图像加灰条避免缩放失真
阅读量:592 次
发布时间:2019-03-12

本文共 1716 字,大约阅读时间需要 5 分钟。

图像加灰条避免缩放失真

在深度学习的训练中,大多需要使用正方形的图片,但是我们平常拍摄的图像大多都是长方形的,所以在缩放图片时,要给图片加灰条,代码很简单:

def letterbox_image(image, size):
iw, ih = image.size
w, h = size
scale = min(w/iw, h/ih)
nw = int(iwscale)
nh = int(ih
scale)

image = image.resize((nw,nh), Image.BICUBIC)new_image = Image.new('RGB', size, (128,128,128))new_image.paste(image, ((w-nw)//2, (h-nh)//2))return new_image

这部分代码是用来加灰条的,iw和ih是原图的行列数,w和h是需要生成的图像的行列数,scale得到的是新图与原图的宽高比,简单想一下,宽高比越小的,变化的范围就越大,那么反之,就要给另外一边补灰条。所以先得到scale这个较小的宽高比,然后对图片进行缩放,new_image是一张灰色的新图,然后将缩放好的图像粘贴到灰色图像上,覆盖掉一部分,那么剩下的部分就是灰条了。

下面附完整的批量加灰条代码:

import osimport numpy as npimport cv2from PIL import Imagedef letterbox_image(image, size):    iw, ih = image.size    w, h = size    scale = min(w/iw, h/ih)    nw = int(iw*scale)    nh = int(ih*scale)    image = image.resize((nw,nh), Image.BICUBIC)    new_image = Image.new('RGB', size, (128,128,128))    new_image.paste(image, ((w-nw)//2, (h-nh)//2))    return new_imagedef search_files(directory):    directory = os.path.normpath(directory)    objects = {}    for curdir, subdirs, files in os.walk(directory):        for file in files:            if file.endswith('.jpg'):                label = curdir.split(os.path.sep)[-1]                if label not in objects:                    objects[label] = []                path = os.path.join(curdir, file)                objects[label].append(path)    return objectsif __name__ == "__main__":    train_samples = search_files('E:\\python\\learning\\tree_learn\\crossFork')    print(train_samples)    for label, filenames in train_samples.items():        for filename in filenames:            img = Image.open(filename)            new_img = letterbox_image(img, (224, 224))            new_img.save(filename)

转载地址:http://hlhtz.baihongyu.com/

你可能感兴趣的文章
mysql 权限整理记录
查看>>
mysql 权限登录问题:ERROR 1045 (28000): Access denied for user ‘root‘@‘localhost‘ (using password: YES)
查看>>
MYSQL 查看最大连接数和修改最大连接数
查看>>
MySQL 查看有哪些表
查看>>
mysql 查看锁_阿里/美团/字节面试官必问的Mysql锁机制,你真的明白吗
查看>>
MySql 查询以逗号分隔的字符串的方法(正则)
查看>>
MySQL 查询优化:提速查询效率的13大秘籍(避免使用SELECT 、分页查询的优化、合理使用连接、子查询的优化)(上)
查看>>
mysql 查询,正数降序排序,负数升序排序
查看>>
MySQL 树形结构 根据指定节点 获取其下属的所有子节点(包含路径上的枝干节点和叶子节点)...
查看>>
mysql 死锁 Deadlock found when trying to get lock; try restarting transaction
查看>>
mysql 死锁(先delete 后insert)日志分析
查看>>
MySQL 死锁了,怎么办?
查看>>
MySQL 深度分页性能急剧下降,该如何优化?
查看>>
MySQL 深度分页性能急剧下降,该如何优化?
查看>>
MySQL 添加列,修改列,删除列
查看>>
mysql 添加索引
查看>>
MySQL 添加索引,删除索引及其用法
查看>>
MySQL 用 limit 为什么会影响性能?
查看>>
MySQL 用 limit 为什么会影响性能?有什么优化方案?
查看>>
MySQL 用户权限管理:授权、撤销、密码更新和用户删除(图文解析)
查看>>