为图像分类创建图像数据集

Question

我想用Pandas和Scikit学习libs创建图像分类器，但是我想拥有自己的图片数据？ 从现在起，我已经使用pandas列作为特征和结果，但是如何才能将图像添加到pandas列（处理它），以便在分类中使用它？你知道吗

我能不能把我的图片添加到熊猫中，处理它们并在分类器中使用它们，或者我需要做一些不同的事情？你知道吗

我有我的.py文件保存在同一个文件夹，我有我的.jpg图像，但我现在不知道如何阅读它们。你知道吗

import pandas as pd
from sklearn.metrics import accuracy_score
from sklearn.ensemble import RandomForestClassifier

images = ['...list of my images...']
results = ['drvo','drvo','cvet','drvo','drvo','cvet','cvet'] 

df = pd.DataFrame({'Slike':images,
                   'Rezultat':results})

print(df)

features = df.iloc[:,:-1]
results = df.iloc[:,-1]

clf = RandomForestClassifier(n_estimators=100, random_state=0)
model = clf.fit(features, results)

prediction_result =  model.predict(features)
print(prediction_result)

accuracy_test = accuracy_score(results, prediction_result)
print(accuracy_test)

Answer 1

我不认为只有猫科动物和熊猫才能阅读图像。我建议您使用numpy数组来表示图像。您可以使用scikit-image和pillow等库。你知道吗

pip install scikit-image

通过以下方式验证安装：

>>> import skimage
>>> skimage.__version__

然后，您可以通过以下方式获得三维numpy阵列图像：

skimage.io.imread(fname)

前两个维度是x和y坐标（其中[0,0]在左上角），第三个维度有颜色通道的值。对于RGB图像，每个[x，y]有3个值，分别对应于红色、绿色和蓝色通道。有关详细信息，请参阅skimage docs。你知道吗