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部分，应用训练模型