1 Star 3 Fork 5

WJG/tflite_train

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
readme.txt 1.95 KB
一键复制 编辑 原始数据 按行查看 历史
WJG 提交于 2020-04-22 13:28 . 添加Serverless云函数部署支持
配置:
conf/sample_config.json:
"image_width": 100, //验证码宽度
"image_height": 50, //验证码高度
"max_captcha": 5, //验证码字符长度
"image_suffix": "jpg", //验证码图像扩展名
"char_set": "23456789abcdefghjkmpqrstuvwxy", //验证码中的所有字符
"cycle_stop": 20000,
"acc_stop": 0.99, //准确率达到该值后结束训练
"cycle_save": 500, //每500轮自动保存一下模型
"enable_gpu": 1, //1 启用GPU训练 0关闭
"train_batch_size": 128,
"test_batch_size": 100
训练结束后会在/model路径下创建模型文件,下载即可。
原项目地址:https://github.com/nickliqian/cnn_captcha
------------------------------
kaggle训练命令(自行修改仓库名称)
--------------------------------------------
!git clone https://gitee.com/LoveWJG/tflite_train.git
cd tflite_train/data/
!wget http://pan.idoo.top/data.zip (下载数据集)
!unzip All.zip -d origin (解压数据集)
cd ../
!python sort.py (划分训练集和验证集)
!python train.py (模型训练)
!python test.py (测试集准确率)
!python save.py (保存模型)
!python tflite.py(量化转换tflite模型)
!python final.py(查看tflite模型识别结果)
!tar -zcvf model.tar.gz model (打包模型以供下载)
!tar -zcvf model2.tar.gz model2(用于生成tflite模型)
!tar -zcvf logs.tar.gz logs(打包日志)
-----------------------------------------------
----------------
打开tensorboard
------------------------------
tensorboard --logdir logs
-------------------
tf1.x代码兼容tf2.0
------------------------------
单个文件
tf_upgrade_v2 --infile train_model.py --outfile train_model_update.py
整个文件夹
tf_upgrade_v2 --intree cnn_captcha --outtree cnn_captcha_new
----------------------
tensorflow模型转tflite
------------------------------------------
常规转换
tflite_convert --saved_model_dir=model2 --output_file=model2.tflite
量化模型(减小体积)
运行项目里的tflite.py即可
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/LoveWJG/tflite_train.git
git@gitee.com:LoveWJG/tflite_train.git
LoveWJG
tflite_train
tflite_train
master

搜索帮助

0d507c66 1850385 C8b1a773 1850385