这个专栏更新各种AI,以及各种有趣的教程,有兴趣的小伙伴可以订阅一下。
先上效果图
首先是数据集的准备
识别部分使用的是cnn算法。
demo如下:
首先是数据集的生成
import os
import random
b = 0
dir = './data/'
#os.listdir的结果就是一个list集合
#可以使用一个list的sort方法进行排序,有数字就用数字排序
files = os.listdir(dir) #获得扩增后图片文件夹路径
files.sort()
#print("files:", files) #创建txt文件用于后续数据储存
train = open('./train.txt', 'w')
val = open('./val.txt', 'w')
a = 0
a1 = 0
while(b < 5):#5是因为5个文件夹 循环5次
label = a #设置要标记的标签
ss = './data/' + str(fi