基于Google object detection API的猫脸识别器

本文介绍了如何基于Google的Tensorflow Object Detection API训练一个猫脸识别器。通过Transfer Learning,利用预训练模型对数百张猫脸数据进行训练,最终得到95%左右准确率的识别器。文章详细阐述了环境搭建、数据准备、训练配置与训练、模型导出和调用的过程,并分享了在训练中遇到的问题和解决方案。
摘要由CSDN通过智能技术生成

近期趁着闲下来整理一下之前做的各种东西。

首先是这么一个猫脸识别器:基于Google提供、Tensorflow框架下的object detection API,只需准备几百张自己的数据集,在Google提供的训练好的模型上进一步训练,可以在短时间内得到一个性能相对较好(准确率95%左右)、最终调用接口也十分方便的自定义识别器。基于已有模型、实现对新物体识别的训练方法叫做Transfer Learning。

个人认为Google这个接口十分良心,直接方便了广大工程师,可以快速地完成自己的特定应用,例如在摄像头中监控特定目标是否出现等等。而且这个识别器既包含分类(Classification)的功能也包含了精确定位位置(Detection)的功能,下一步大概就是Segmentation了吧(笑)。

再说回猫脸识别。调查猫脸识别的时候,我们发现CSDN上倒是有基于opencv的代码,但是在Cats v.s. Dogs的数据集上跑了一下效果并不好,并且各种参数十分莫名根本不知道该怎么调。该数据集本来是一个猫脸狗脸识别大赛的数据集,对于我们的应用刚好合适。Google的API有整只猫的识别,但是没有猫脸和猫眼睛的识别,于是就想到了在已有模型上进行Transfer Learning,可以快速达到目标。

主要参考了以下内容:

[1] Google自己github上的说明,此说明写的非常详细,只看这份说明就可以完成所有的操作。

https://github.com/tensorflow/models/tree/master/research/object_detection.

[2] 英文不够好导致不能清楚理解的朋友可以跟我一一样参考一下这篇中文博客~一篇讲的还可以但不够仔细的linux上的实现。

https://blog.csdn.net/c2a2o2/article/details/78436735.

[3] 原博文找不到了,在原模型基础上学习识别Racoon(小浣熊!),github上的代码架构十分清晰,但是README写的十分简略。

  • 0
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
TensorFlow Object Detection API 是一个开源项目,它提供了一系列基于 TensorFlow 的工具和库,用于实现目标检测任务。对于 macOS 系统,我们可以通过以下步骤来使用 TensorFlow Object Detection API: 1. 安装 TensorFlow:在 macOS 上安装 TensorFlow 是使用 TensorFlow Object Detection API 的前提。你可以通过 pip 命令进行安装,例如在终端中执行 `pip install tensorflow`。 2. 下载 TensorFlow Object Detection API:打开终端并导航到适合你的工作目录中,然后使用 git 命令来克隆 TensorFlow Object Detection API 的 GitHub 仓库,例如执行 `git clone https://github.com/tensorflow/models.git`。 3. 安装依赖项:进入克隆的模型目录中,找到 research 文件夹并进入。然后运行 `pip install -r object_detection/requirements.txt` 命令来安装所需的依赖项。 4. 下载预训练模型:在 TensorFlow Object Detection API 中,我们可以使用预训练的模型来进行目标检测。你可以从 TensorFlow Model Zoo 中下载适合你任务的模型,并将其解压到你的工作目录中。 5. 运行实例代码:在 research/object_detection 目录中,你可以找到一些示例代码,用于训练、评估和使用目标检测模型。可以通过阅读这些示例代码并根据自己的需求进行修改。例如,你可以使用 `python object_detection/builders/model_builder_tf2_test.py` 命令来运行一个模型的测试。 以上是在 macOS 上使用 TensorFlow Object Detection API 的基本步骤,你可以根据你的具体需求进行更多的深入研究和调整。希望这些信息能帮助到你!
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值