KIWI-ST / kiwi.brain

resolving the trace network generation by using deep learning

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

kiwi.brain

deep learning to resolve trace network generation / trajectory correction ...

description

使用resnet对样本进行训练,得到的结果可以在 kiwi.server中,对图像进行分类操作

features

1.train.records 构建了分类结果为11类的样本
2.样本图片的尺寸为 10x10,通道为1
3.样本尺寸尺寸可自定义修改

usage

1.构建records样本,使用 src/utils/build_records.py 将样本图片和label构建train.tfrecords和eval.tfrecords
2.使用src/brains/resnet_main.py 训练模型,注意tf.Flags修改相关参数,常修改的参数是样本图片的规格
3.训练后的checkpoint可以使用utils/freeze_object生成 .pb文件,配合kiwi.server 的host.image.ui部分,应用训练模型

About

resolving the trace network generation by using deep learning

License:GNU General Public License v3.0


Languages

Language:Python 100.0%