一个简单的图像分类项目练习(一) 开端

        有了之前的CIFAR-10项目实践,可以模仿它的数据结构和学习过的网络结构,创建一个简单的图像分类项目作为练习和总结。

        作为一个图像分类项目,应该包括:训练图像集、测试图像集、加载和训练以及测试的脚本、训练好的模型、调用模型的脚本。如果时间和精力足够,增加前端显示界面。
        

 一、打开pycharm,新建项目:image_classification,文件目录结构如下:

        image文件夹:存放图像文件。其下的train、test、error文件夹分别存放训练、测试和测试错误的图像文件。
        lib文件夹:存放一些自定义的有可能在别的项目中用到的通用库。
        model文件夹:存放训练好的模型。
        script文件夹:存放功能脚本。

二、在lib和script文件夹下分别创建__init__.py文件,内容为空,这样,文件夹就会被视为一个python包,可以使用从python包内import脚本的方法来调用文件夹内的脚本。

        当文件夹被视为包,图标与普通文件夹的图标是有区别的。

三、准备数据:仍然使用CIFA10数据集。为更贴近实际的工程应用,直接使用解压好和分类好的图像文件夹。拷贝图像到image文件夹:

所有相同分类的图像文件放在以分类命名的文件夹内:

猜你喜欢

转载自blog.csdn.net/xulibo5828/article/details/143230646