图像分类预处理

本文介绍了图像分类预处理的两种方法:一是将训练集图片分为x_train和y_train,可直接使用fit方法训练;二是将图像按类别保存到不同文件夹,并利用ImageDataGenerator生成训练和开发集。内容涉及np.where的运用和shutil工具的使用。
摘要由CSDN通过智能技术生成

对于训练集图片分类记录在csv文件中的情况:

有两种处理方式:

一、将训练集图片分为x_train和y_train:

#读入数据
labels=pd.read_csv('labels.csv')
sample_submission = pd.read_csv( 'sample_submission.csv')
#进行one-hot编码,并转为np.array
targets_series=pd.Series(labels['breed'])
one_hot=pd.get_dummies(targets_series,sparse=True)
one_hot_labels = np.asarray(one_hot)
#用opencv读取图片
img_size=299
x_train=[]
y_train=[]
x_test=[]
i=0
for f,breed in tqdm(labels.values):
    img=cv2.imread('train/{}.jpg'.format(f))
    label=one_hot_labels[i]
    x_train.append(cv2.resize(img,(img_size,img_size)))
    y_train.append(label)
    i+=1
for f in tqdm(sample_submission['id'].values):
    img=cv2.imread('test/{}.jpg'.format(f))
    x_test.append(cv2.resize(img,(img_size,img_size)))
#将图片处理为已有模型所需的输入
y_train_raw = np.array(y_train, np.uint8)
x_train
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值