AI识别图片中是小孩还是大人
AI识别图⽚中是⼩孩还是⼤⼈关于奉献的演讲稿
最近使⽤很多软件的时候会提⽰可以选择开启青少年模式，这些⼤⼚在保护青少年有所⾏动是好事，因为现在很多⼩朋友都开始玩⼿机。我就想起来能不能有⼀个算法去识别图⽚中是⼩孩或者是⼤⼈，这对计算机视觉来说不是⼀个复杂的问题，这是⼀个典型的分类问题。
女儿思念天堂的妈妈说说在上⼀篇⽂章中图像分类实践展⽰了使⽤docker安装⼀个cpu的模型训练环境，今天我们在windows上安装⼀个GPU的版本。
万事开头难
需要安装
1. anaconda3
python的科学计算环境
2. CUDA
GPU程序包
3. cudnn
朱丹直播回应口误神经⽹络的包，需要到官⽹右键查看下载链接，最好⽤迅雷下载
4. pytorch
凤尾菇栽培技术最流⾏的AI(⼈⼯智能)⼯具
配置清华的源
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
5. fastai
基于上⾯的⼯具开发的快速开发深度学习算法程序的⼯具
conda install -c fastai fastai
准备数据
随便在⽹上下载了⼏⼗张照⽚，分别是百度图⽚和bing图⽚搜索的
训练数据 ⽂件源只有22张照⽚
设置了训练数据的⽬录，修复中⽂label显⽰的问题。
准备测试数据
标签函数需要⾃定义
查看测试数据
接下来基于上⾯的数据进⾏模型训练，训练的过程中打印错误率。这⾥基于resnet34算法，属于迁移学习，还使⽤的最新的训练⽅法。并没有进⾏调优。
训练
测试数据
我们使⽤新测试的数据以便区别训练数据和验证数据，使⽤模型没有见过的图⽚。
测试数据集看看实际的效果
测试⼀
测试⼆
看起来还不错的样⼦。
使⽤的电脑是普通的笔记本
GPU截图 GPU MX150
安装机器学习的环境还⽐较快，花费了⼀两个⼩时（希望只安装⼀次）。GPU的内存⽐较⼩，需要调⼩每次训练batch的size，如果内存溢出需要重新启动kernel。
跑fastai的最简单⽰例竟然需要七分钟，看来还是要免费的GPU。
参考资料：
win10安装CUDA和cuDNN的正确姿势
为什么动物要冬眠
安装pytorch
拍一拍怎么弄
fastai 版本问题

版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198，我们将在24小时内删除。