深度学习值pytorch实战计算机视觉——第七章猫狗数据集预处理的代码

因为书上给出的加载数据集是已经经过预处理的，并不是直接从网上之间下载的数据集。所以需要对其进行预处理。下面是进行预处理的代码。主要是将猫狗的图片放到对应的数据集中，所以这个代码的主要功能就是将对应的图片移动到对应的文件夹中，但不改变其文件名。

import os
import shutil
file_dir="/home/zou/code/learn/pytorch/chapter-7/kaggle_DogsVSCats/test1"
cat="/home/zou/code/learn/pytorch/chapter-7/DogsVSCats/valid/cat"
dog = "/home/zou/code/learn/pytorch/chapter-7/DogsVSCats/valid/dog"
def file_name(file_dir):
    str=[]
    i=0
    for root, dirs, files in os.walk(file_dir):
        root=root
        dirs=dirs
        str=files
    for i in str:
        if "cat" in i:
            shutil.move(root+"/"+i,cat+"/"+i)
        else:
            shutil.move(root+"/"+i,dog+"/"+i)
file_name(file_dir)

其中file_dir是原始猫狗图片所在的文件夹，cat和dog则分别是处理之后猫狗图片应在的文件夹。