图像识别实战(一)----如何根据已有图片生成测试和训练信息并保存到文件
提前知道
本实验为手势识别,具体图片可以参考下图,经过训练我们得知,标签为5.
总体目标
Dataset文件夹下面有0-9十个文件,其中有几十张代表文件夹数值的手势图片。
我们的目标便是将Dataset文件夹下面的图片进行训练,最后得出模型进行预测。
本次目标
将Dataset下面的所有图片的路径及其标签保存到两个文件train.list&test.list中,便于模型训练时调用。
1、导入库
import os
import time
import random
import numpy as np
from PIL import Image
import matplotlib.pyplot as plt
import paddle
import paddle.fluid as fluid
import paddle.fluid.layers as layers
from multiprocessing import cpu_count
from paddle.fluid.dygraph import Pool2D,Conv2D
from paddle.fluid.dygraph import Linear
导入库的时候可以导入多余的库,所以可以放心导入。
2、生成保存信息的文件
data_path = '/home/aistudio/data/data23668/Dataset'
#每个人的数据都可能保存