教程 | 猫狗识别 - AI图片分类的样例

最新推荐文章于 2024-09-15 15:58:40 发布

iFlyAI

最新推荐文章于 2024-09-15 15:58:40 发布

阅读量2.9k

点赞数

分类专栏：代码样例文章标签： TensorFlow Keras Pytorch FlyAI AI

本文链接：https://blog.csdn.net/iFlyAI/article/details/85054592

版权

代码样例专栏收录该内容

11 篇文章 1 订阅

订阅专栏

猫狗识别 - AI图片分类的样例

更多项目请查看 https://www.flyai.com

项目官方网址

该数据有猫和狗两类图片，一共有25000张图片，猫狗各占一半，可以用来做图片分类的入门数据。

样例代码说明

`app.yaml`

是项目的配置文件，项目目录下必须存在这个文件，是项目运行的依赖。

`processor.py`

样例代码中已做简单实现，可供查考。

处理数据的输入输出文件，把通过csv文件返回的数据，处理成能让程序识别、训练的矩阵。

可以自己定义输入输出的方法名，在app.yaml中声明即可。

    def input_x(self, image_path):
        '''
        参数为csv中作为输入x的一条数据，会被Dataset类中的next_batch()方法多次调用。
        :params: 输入的数据列表
        :return: 返回矩阵
        '''
        pass

    def input_y(self, label):
        '''
        参数为csv中作为输入y的一条数据，会被Dataset类中的next_batch()方法多次调用。
        :params: 数据标签列表
        :return: 返回矩阵
        '''
        pass
    def output_y(self, data):
        '''
        验证时使用，把模型输出的y转为对应的结果
        :param data: 预测返回的数据
        :return: 返回预测的标签
        '''
        pass

`main.py`

样例代码中已做简单实现，可供查考。

程序入口，编写算法，训练模型的文件。在该文件中实现自己的算法。

通过dataset.py中的next_batch方法获取训练和测试数据。
# 数据获取辅助类
dataset = Dataset()
x_train, y_train, x_test, y_test = dataset.next_batch(BATCH)
通过model.py中的save_model方法保存模型
# 模型操作辅助类
model = Model(dataset)
model.save_model(YOU_NET)
如果使用PyTorch框架，需要在net.py文件中实现网络。其它用法同上。

`model.py`

样例代码中已做简单实现，可供查考。

训练好模型之后可以继承flyai.model.base包中的base重写下面三个方法实现模型的保存、验证和使用。

   def predict(self, path, name, **data):
        '''
        使用模型
        :param path: 模型所在的路径
        :param name: 模型的名字
        :param data: 模型的输入的一个或多个参数
        :return: 
        '''
        pass

    def evaluate(self, path, name):
        '''
        验证模型
        :param path: 模型的路径
        :param name: 模型的名字
        :return: 返回验证的准确率
        '''
        pass

    def save_model(self, network, path=MODEL_PATH, name=MODEL_NAME, overwrite=False):
        '''
        保存模型
        :param network: 训练模型的网络
        :param path: 要保存模型的路径
        :param name: 要保存模型的名字
        :param overwrite: 是否覆盖当前模型
        :return:
        '''
        self.check(path, overwrite)

`predict.py`

样例代码中已做简单实现，可供查考。

对训练完成的模型使用和预测。

`path.py`

可以设置数据文件、模型文件的存放路径。

`dataset.py`

该文件在FlyAI开源库的flyai.dataset包中，通过next_batch(BATCH)方法获得x_train y_train x_test y_test数据。

FlyAI开源库可以通过pip3 install -i https://pypi.flyai.com/simple flyai 安装。