基于Keras的CNN框架

以下内容来自我的github仓库,具体框架的代码也可以在仓库中找到,更加欢迎各位一起帮助我修改。

[法师传送门]https://github.com/AngularLock/DeepLearningFramework

A Deep Learning Framework based on Keras

Keras搭建的深度学习包 类比于MatConvNet

Tutorial

Keras is a high-level neural networks library, written in Python and capable of running on top of either TensorFlow or Theano.Below,I’ll show my framework which is to solve cifar10.With the framework,you can focus on the parameters of the neurn network

Keras是一个简单好用的python深度学习框架,后端可以选择Tensorflow和Theano.以下用cifar10分类来介绍这个已经搭建好的包,以此,你只需要更加关注CNN参数的调整

implementation function 实现功能

  • load data, buid model, train model
  • save better model_weights,you can continue training after a pause
  • at the end of every epoch will show loss&error curve

  • 读取数据,生成模型,训练模型

  • 网络可以实时保存权值,在终端之后依旧可以继续运算。
  • 在训练时,每一个epoch结束后会显示 loss 和 error曲线

1. The structure 网络结构

  • data_generator
  • model_setup
  • callbackfn
  • main

1.1 data_generator

to load and normalize raw data,this module returns 4 variable: x_train,t_train(label),x_test,t_test(label).

这个模块用于加载和规范化原始数据文件,有4个返回值:x_train,t_train(类标),x_test,t_test(类标)

1.2 model_setup

you can build your model in this module.Return the model as the returned vlue.

在此模块 你可以自由搭建CNN网络。 此模块返回值为model

1.3 callbackfn

this module is built to show the loss & error curve while training.

此模块可以实现在网络训练时 实时画出loss & error曲线

1.4 main

this file like the main function in c or other program language.all the modules above will be called in this file.

主文件,调用以上所有模块,完成主要流程:load data–build model–train model

2. what to do next 后续版本

2.1 Bug Fix 错误修改

  • sometimes crashed when ploting loss&error curve
  • sometimes warnings shows when building or training model

  • 在画loss&error曲线时会无响应

  • 在搭建和训练模型时会弹出某些警告

2.2 to be done 后续功能

  • prediction module

  • 预测 测试 模块

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值