Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings

bubbliiiing/Transfer-Learning

Open more actions menu

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
25 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Transfer-Learning-迁移学习思想Keras当中的实现


目录

  1. 所需环境 Environment
  2. 仓库内容 WhatsIn
  3. 文件下载 Download

所需环境

tensorflow-gpu==1.13.1
keras==2.1.5

仓库内容

仓库保存了三个例子,分别是:

  • VGG16训练分类网络-利用dense全连接完成分类-猫狗大战
  • MobileNet训练分类网络-利用GlobalAveragePooling全局池化完成分类-猫狗大战
  • Resnet50-Segnet训练语义分割网络-区分斑马线

这三个例子均使用了迁移学习的思想,主干部分提取出来的特征是通用的,所以我们可以不训练主干部分先,因此训练部分分为两步,分别是冻结训练和解冻训练,冻结训练是不训练主干的,解冻训练是训练主干的。 由于训练的特征层变多,解冻后所需显存变大。

文件下载

猫狗数据集:链接: https://pan.baidu.com/s/1TqmdkJBY49ftg19tRK2Ngg 提取码: htxf
斑马线数据集:连接:链接:https://pan.baidu.com/s/1uzwqLaCXcWe06xEXk1ROWw 提取码:pp6w

About

这是一个有关迁移学习的仓库,在这里可以看到迁移学习的各种用法。

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

Morty Proxy This is a proxified and sanitized view of the page, visit original site.