上面代码将为每个类下载2000张图像,可以编辑最后一个“if”条件来更改下载图像的个数。
数据准备
我们下载的文件夹可能包含其他类型的文件,所以首先必须删除不需要的类型的文件。...image_exts = ['jpeg','.jpg','bmp','png']
path_list = ['drawings','hentai','neutral','porn','sexy']...因为使用TensorFlow框架所以需要判断是否被TensorFlow支持,所以我们这里加一个判断:
import tensorflow as tf
os.chdir('{data-set} directory...', '.png', '.gif', '.bmp']:
print('Unsupported image format:', ext)
os.remove...','test2.jpeg','test3.jpeg','test4.jpeg','test5.jpeg']
print_classes(li,model)
看结果还是可以的。