PAN.pytorch icon indicating copy to clipboard operation
PAN.pytorch copied to clipboard

cpp的pse有些问题

Open xxlxx1 opened this issue 5 years ago • 3 comments

某些图片在cpp的pse过程会卡住,没报错一直卡住,改成pypse就可以正常运行。pypse代码里有个bug,25行应该改成for i in range(label_values): 之前是for i in label_values:,label_values是个int

xxlxx1 avatar Dec 04 '19 01:12 xxlxx1

找到问题了,是因为cpppse里面的结果矩阵类型是uint8所以只支持256个框以内的,我的图片框比较多,就会有问题, //初始化结果 auto res = py::array_t<int32_t>(pbuf_text.size); 改成int32就行了

xxlxx1 avatar Dec 05 '19 03:12 xxlxx1

只有我在改了后,出现段错误的情况么。 按楼上说的,改为int32_t之后,训练没问题,但是预测时候基本上所有图片都出现段错误的情况,改回去就好了。。。

ww451575464 avatar Jun 07 '20 08:06 ww451575464

某些图片在cpp的pse过程会卡住,没报错一直卡住,改成pypse就可以正常运行。pypse代码里有个bug,25行应该改成for i in range(label_values): 之前是for i in label_values:,label_values是个int

pypse 怎么使用,我看 postprocessing init.py 使用的是 from .pse import pse_cpp, get_points, get_num, 这里应该是用到了 pse.so文件吧,如何直接运行 pypse 不用cpp的代码呢

Zhang-O avatar Oct 09 '20 07:10 Zhang-O