小项目——自制画风数据集

最新推荐文章于 2024-09-13 21:33:31 发布

爱吃鱼的小刘

最新推荐文章于 2024-09-13 21:33:31 发布

阅读量1.1k

点赞数 1

分类专栏：小项目

本文链接：https://blog.csdn.net/qq_38484259/article/details/84791079

版权

本文介绍了作者如何从爬取的涂鸦、油画和素描图片中创建一个干净的数据集。首先，统一图片尺寸为32x32，接着通过文件名前缀进行分类，最后进行数据归一化并处理不同通道数的问题，特别是处理灰度图，以创建一个32×32×3的合格图像数据集。

摘要由CSDN通过智能技术生成

承接上篇爬取了想要的数据集之后，先将图片的尺寸统一。(我自己爬取了涂鸦，油画，素描这3类图片）

关于涂鸦数据集爬下来的图片，其中混入了一些妹子和西海岸黑人黑帮文化的图片。(为什么我也不知道。。。)注意清除，数据集的好坏很关键！

resize

先统一图片格式32*32

# coding=utf-8
from PIL import Image
import os.path
import glob
def convertjpg(jpgfile,outdir,width=32,height=32):
    img = Image.open(jpgfile)
    try:
        new_img = img.resize((width,height),Image.BILINEAR)
        new_img.save(os.path.join(outdir,