/dogs_cats

Udacity P7 Project

Primary LanguageJupyter Notebook

Udacity Machine Learning Engineer Nanodegree

Capstone Proposal - Kaggle Cats Vs Dogs

Introduction

通过训练深度卷积神经网络来识别图中的主体为猫或狗。 在该项目中，尝试使用迁移学习的方法，合并多个预训练模型来帮助我们在数据量有限的情况下训练出识别能力优秀的模型。

Install

该项目使用Python 3.5并且需要以下Library

  • Numpy
  • Matplotlib
  • Keras
  • scikit-learn
  • HyperOpt(其中networkx==1.11,不可使用最新版)
  • tqdm

训练、测试数据

https://www.kaggle.com/c/dogs-vs-cats/data.