分别利用KNN与Random Forest 算法完成手绘Fashion-MNIST 10类简笔画的识别

1.Random Forest

1.1 数据准备

将下载好的Fashion-MNIST数据集中的trian集合转化为jpg格式并生成label的train.txt

import torchvision.datasets.mnist as mnist

root="fashion_mnist/"
train_set = (
    mnist.read_image_file(os.path.join(root, 'train-images-idx3-ubyte')),
    mnist.read_label_file(os.path.join(root, 'train-labels-idx1-ubyte'))
        )

根据训练集的label种类手绘test集的jpg文件