Git Product home page Git Product logo

cnn's Introduction

CNN

Training a convolutional neural network using its own picture dataset

Run train.py directly

The running parameters of the network are saved in the CK+ folder. If you change the name, retraining the thing.ckpt will be overwritten.

Explain:Since the previously used dataset (CK+Expression Database) has a problem: the dataset has both grayscale and color maps, so the previous test code reports an error from time to time. This test code is supplemented by the follow-up. In order to simplify the training and facilitate the test, three types of pictures are randomly crawled on the network as the training set and the test set.

When actually use your own datasets, just put the datasets together with codes in a folder, modify the relevant path in the code.

More specific can refer to the blog: Https://blog.csdn.net/jesmine_gu/article/details/81155787

For reference only, there are errors welcome to correct me!

直接运行train.py就可以

网络的运行参数保存在CK+文件夹里面,不改名字重新训练thing.ckpt会被覆盖

说明,由于之前用的数据集(CK+表情数据库)有一点问题:数据集中既有灰度图也有彩色图,所以先前的测试代码不时报错。现在补上测试代码,为了简化训练方便测试,通过在网上随机爬取三类图片作为训练集和测试集。

实际换用自己的数据集时,只要把是数据集跟代码放在一个文件夹里,修改训练以及测试代码中的相关路径就可以了。

更具体的可以参见博客: 完整实现利用tensorflow训练自己的图片数据集 https://blog.csdn.net/jesmine_gu/article/details/81155787

仅供参考,有错误欢迎指正!

cnn's People

Contributors

nangc avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.