用到的github网址:https://github.com/argman/EAST
需下载里面的icdar2015数据集,我只下了training set和对应的gt。今天先讲训练部分~
环境同样是本机cpu环境下。。 没有gpu太难了
一、数据处理部分
由于程序里写的ground truth的名字是例如img_1.txt,而实际上下载下来的是
所以要么改代码,要么改文件名。 我选择了改代码。在icdar.py里604行,改成这样就好啦
txt_fn ="/Users/vikki/Downloads/ch4_training_images/gt_"+ str(im_fn.split("/")[-1].split(".")[0]+".txt")
然后再去multigpu_train.py里根据情况改tf.flags.DEFINE_string参数,比如学习率,checkpoint_path啥的。还有icdar.py里training_data_path路径。
然后就可以运行multigpu_train.py了
cpu太慢了 我只用了1000张图片训训看
然后等着就行了,会存在你指定的文件夹里。
在此期间可以利用tensorboard实现可视化
tensorboard --logdir=east_icdar2