人名分类器实战项目
0 说明
本项目对前一个博客内容2.5(cpu version) 人名分类器实战项目(对比RNN、LSTM、GRU模型)的区别在于两点:
- (1)采用gpu
- (2)对项目代码进行工程化管理
第一点不用多讲,就是将模型和数据传入到gpu设备上即可,第二个是由于受到项目开发的影响,按照功能对代码进行工程模块化管理,此方式在编写和调试过程中也非常适用,后期的维护也十分方便(前面cpu版本的代码都在一个文件上,要进行微调就可能执行整个文件代码,不友好)
1 工程项目设计
整个工程项目设计按照如下的方式进行,见下图
其中data文件夹用于放置原始的文本数据、nameclass_cpu.py文件存放上一个博客中所有的代码(即将整个工程项目的代码都存放在一个文件汇总)
Nameclassed文件夹存放本博客中内容,其中img和models分别存放图片和训练好的模型文件,剩下的py文件按照对应的功能进行代码设计。
2 数据预处理data_processing
<