surfzjy / east-caffe Goto Github PK

同学你好，感谢你开源EAST-caffe，很漂亮的工作。
但是我在调试的时候，用caffe作为框架，发现最终层（f_score和geo_concat）的值全部都是NAN。于是我逐层输出每一层的feature map，发现conv_11_pw_hswish开始的输出中包含INF，并且之后的层里INF越来越多，导致最终无法得到想要的结果。

我的系统是ubuntu16，caffe是weiliu89/SSD的caffe，opencv版本4.1.1，RELU6层直接用普通的RELU代替了。
请问下你的RELU6层自己重新实现的，还是也用的普通RELU代替？

Traceback (most recent call last):
File "train.py", line 27, in
train(args.initmodel, args.gpu)
File "train.py", line 15, in train
solver = caffe.AdamSolver('solver.prototxt')
File "/home/csy/TextField/examples/EAST-caffe/pylayerUtils.py", line 49, in reshape
self.data, self.score_map, self.geo_map = self.load()
ValueError: too many values to unpack
看了下def get_whole_data（）函数，return了5个参数
if len(images) == batch_size:
return images, image_fns, score_maps, geo_maps, training_masks
images = []
image_fns = []
score_maps = []
geo_maps = []
training_masks = []
这里我去掉了training_masks和image_fns，images中的任意一个,出现这个错误，所以返还的参数是否有误.
*Traceback (most recent call last):
File "train.py", line 27, in
train(args.initmodel, args.gpu)
File "train.py", line 15, in train
solver = caffe.AdamSolver('solver.prototxt')
File "/home/csy/TextField/examples/EAST-caffe/pylayerUtils.py", line 45, in reshape
top[0].reshape(self.data.shape)
AttributeError: 'list' object has no attribute 'shape'

@SURFZJY What is the code status?
will you add documentation and pre-trained models?

I run my model by demo.py
The model is trained by caffe
The result seems right when I choose dnn api ,but it is wrong when I swap to caffe api.
How to solve this problem?

Hi there,

I'm afraid I only speak English. I would like to download your trained models but I can't understand the download page. How can I do that (apart from learning a new language)?

Thanks.

I have trained this model in ic15 with code in this repository.
After 69100 iteration training, the loss is reduced to 0.00333686.

I1024 22:11:46.818552 4375 sgd_solver.cpp:105] Iteration 69180, lr = 0.001
I1024 22:11:53.286249 4375 solver.cpp:218] Iteration 69190 (1.54621 iter/s, 6.46743s/10 iters), loss = 0.00333686
I1024 22:11:53.286276 4375 solver.cpp:237] Train net output #0: Loss_rbox = 0.00646377 (* 1 = 0.00646377 loss)
I1024 22:11:53.286281 4375 solver.cpp:237] Train net output #1: Loss_score = 0.00405699 (* 0.01 = 4.05699e-05 loss)

However, The performance of trained model is pool even in train set.

这项目太棒了，给学习文本定位技术一个非常好的源码可学习。我在ubuntu 16.04 LTS, python 2.7，ssd-chuanqi305的caffe环境下，编译geo_map_cython_lib, caffe, pycaffe成功，运行出现问题，错误代码如下

$ python train.py --dataset ic15 -gpu 0
 File "/home/link/EAST-caffe/icdar.py", line 13, in 
    from geo_map_cython_lib import gen_geo_map
ImportError: No module named geo_map_cython_lib

请问是运行环境不一样吗，还是我哪里出现了错误?

你好，我是用icdar15数据集进行训练，每次都卡在 conv1_1, 很长时间后，就自动退出了，请问你遇到这样的问题了么？