基于tensorflow实现的猫狗识别
猫狗识别
免责声明:本项目不适合用于毕设,可以用于充当课设模板或者日常学习使用。
项目使用tensorflow、cudnn、anaconda和pycharm等都需要下载。
其中,
- tensorflow下载安装可以阅读:tensorflow下载,如果过程遇到问题(一般肯定会遇到吧,可以看一下关于使用tensorflow_gpu遇到的问题)
- cudnn可以看:cuda与cudnn下载(tensorflow-gpu)
- anaconda:Anaconda下载安装
- pycharm:直接通过官网PyCharm官网下载安装,挺简单的,主要是python反而麻烦,但是你别告诉我:都想练项目或是做课设了,python没搞定,平时实验不会都是Ctrl+C and Ctrl+V吧!
- 训练集与测试集下载地址:猫狗数据集全,test1就是test,test_img可以从别的地方或者从训练集和测试集中抽取
训练时用train.py,迭代次数可以改,看你自己电脑的能力(显卡),会保存成CatsVsDogs_model.h5。
然后再gui.py中进行识别。
如果你上面的都准备好了,那你可以开始项目!!!
项目地址:cat-dog-recognition
加油吧😘😘😘!!!