当前位置: 首页 > 资讯 > 正文

快速训练集预处理-调整比例+AI抠图

来源:哔哩哔哩    时间:2023-04-04 00:05:35

之前处理珐姐的图片时,被AI抠图网站登录机制恶心到了,而且网站不支持批量,一张张处理太麻烦了,所以当时花了十几分钟写了代码。

后来整理的时候发现还是写得挺漂亮的,丢了可惜,所以优化了一下放出。

以下是通过添加白边的方式调整比例:


(资料图片)

代码主要用于图片白边处理,以下是各个函数的功能:

添加白边,输入图片路径、输出图片路径和纵横比,输出白边后的图片。

获取目录下所有图片路径。

重置并创建文件夹。

处理图片,输入图片文件夹路径和最大任务数,输出在文件夹./edge下。

可通过相关注释调整图像比例或修改白边为其他颜色。

其中,任务数可以通过调整 max_tasks参数进行设置。

以下是AI抠图,利用了remove.bg的api,百度的太烂了,remove.bg提供了免费的50次,但是注册非常容易,随便搞几百个网易邮箱约等于无限用。

https://www.remove.bg/dashboard#api-key 申请,普通用户有50次白嫖机会

X 关闭

最近更新

Copyright ©  2015-2022 大众字画网版权所有  备案号:豫ICP备20014643号-14   联系邮箱: 905 14 41 07@qq.com