1 Star 3 Fork 5

WJG / tflite_train

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README
配置:
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即可

空文件

简介

tensorflow通用端到端验证码识别模型训练保存部署方案 展开 收起
Python
取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
Python
1
https://gitee.com/LoveWJG/tflite_train.git
git@gitee.com:LoveWJG/tflite_train.git
LoveWJG
tflite_train
tflite_train
master

搜索帮助