search

WenmuZhou / PAN.pytorch

A unofficial pytorch implementation of PAN(PSENet2): Efficient and Accurate Arbitrary-Shaped Text Detection with Pixel Aggregation Network
Apache License 2.0
418 stars 112 forks source link

特别密集的文本会有内存溢出的情况 #23

Open Brave731 opened 4 years ago

Brave731 commented 4 years ago

特别密集的文本优化内存溢出的情况，导致程序挂掉。

zhangjunyi1225054736 commented 4 years ago

我好像也遇到了这个问题，请问有办法解决吗

zhangjunyi1225054736 commented 4 years ago

我找到解决办法了，在c++的代码里改一下（pse那部分）

Brave731 commented 4 years ago

具体改的那部分可以说说吗，队列的弹入弹出？我之前监控程序挂掉是显存溢出，这个属于后处理应该和显存没关系吧

zhangjunyi1225054736 commented 4 years ago

之所以会有内存溢出，是因为pse.cpp中有个while(!q.empty())，在聚类的过程中，有些像素反复被添加进队列，导致死循环，q中像素越来越多，最后就挂了（不是显存溢出，是内存溢出）。解决办法很简单，就是标记一下，每个像素只能被添加进队列一次。

delamoerxy commented 4 years ago

@zhangjunyi1225054736 我也遇到了相同的问题，在预测一些文字比较多的图片的时候内存会溢出。但是我对c++不太熟，不知道怎么修改。请问具体是修改pse.cpp中的哪个地方呢？

zhangjunyi1225054736 commented 4 years ago

@zhangjunyi1225054736 我也遇到了相同的问题，在预测一些文字比较多的图片的时候内存会溢出。但是我对c++不太熟，不知道怎么修改。请问具体是修改pse.cpp中的哪个地方呢？ 关注我的知乎MoonSmile，我私发代码给你

delamoerxy commented 4 years ago

@zhangjunyi1225054736 太感谢你了，我已经在知乎关注你了

pekinghk commented 4 years ago

我也遇到了这个问题，请问后来解决了么

Zhang-O commented 4 years ago

@zhangjunyi1225054736 我也遇到了相同的问题，在预测一些文字比较多的图片的时候内存会溢出。但是我对c++不太熟，不知道怎么修改。请问具体是修改pse.cpp中的哪个地方呢？ 关注我的知乎MoonSmile，我私发代码给你

Share your solution here will be better than the wa y you mentioned above. We hope the communications among us is as efficiency as possible!

PanFei748 commented 4 years ago

@zhangjunyi1225054736 太感谢你了，我已经在知乎关注你了

你好，可以也分享一份给我吗。。。不胜感谢：）

PanFei748 commented 4 years ago

具体改的那部分可以说说吗，队列的弹入弹出？我之前监控程序挂掉是显存溢出，这个属于后处理应该和显存没关系吧

请问你解决这个问题了吗。。。if yes：可以分享一下吗。。。不胜感激：)

Brave731 commented 4 years ago

后处理函数中，pse.cpp 82行的循环种，定义一个一维数组，判断是否遍历过。 image

PanFei748 commented 4 years ago

emmm，does not work....方便发一下您的pse.cpp文件吗。。不胜感激！

------------------ 原始邮件 ------------------ 发件人: "WenmuZhou/PAN.pytorch" <notifications@github.com>; 发送时间: 2020年7月17日(星期五) 上午10:41 收件人: "WenmuZhou/PAN.pytorch"<PAN.pytorch@noreply.github.com>; 抄送: "潘飞"<764455491@qq.com>;"Comment"<comment@noreply.github.com>; 主题: Re: [WenmuZhou/PAN.pytorch] 特别密集的文本会有内存溢出的情况 (#23)

后处理函数中，pse.cpp 82行的循环种，定义一个一维数组，判断是否遍历过。

— You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe.

PanFei748 commented 4 years ago

后处理函数中，pse.cpp 82行的循环种，定义一个一维数组，判断是否遍历过。 image

您好，请问temp_map是如何定义的啊，不熟悉C++，不好意思啊~~~

Brave731 commented 4 years ago

image

PanFei748 commented 4 years ago

emmm,还是在那一步会内存溢出。方便发一份你的pse.cpp吗~

------------------ 原始邮件 ------------------ 发件人: "WenmuZhou/PAN.pytorch" <notifications@github.com>; 发送时间: 2020年7月17日(星期五) 中午1:45 收件人: "WenmuZhou/PAN.pytorch"<PAN.pytorch@noreply.github.com>; 抄送: "潘飞"<764455491@qq.com>;"Comment"<comment@noreply.github.com>; 主题: Re: [WenmuZhou/PAN.pytorch] 特别密集的文本会有内存溢出的情况 (#23)

— You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe.

archwolf118 commented 4 years ago

后处理函数中，pse.cpp 82行的循环种，定义一个一维数组，判断是否遍历过。 image

是不是应该

if(temp_map[tmpy*w+tmpx])
    continue;
temp_map[tmpy*w+tmpx] = true;

if 后面多了一个！

code4101 commented 3 years ago

参考改法：https://github.com/code4101/PAN.pytorch/pull/1/commits/41f517042b1742ed02f3d80b5c5c9bbc3d9fd880 测试了下，对整体运行速度没影响，但增加了些内存开销。 （一些操作失误，重复commit，不好意思~~）