EraseNet
EraseNet copied to clipboard

Can you provide a pre-trained model?Thanks~
Hi,thank u for your wonderful work and the open source code. do u have the plan to provide the pretrain model?
Getting below error under pytorch version >1.4 . but it's good at pytorch 1.4. RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation:...
测试结果异常
下载模型测试结果怎么会是这样的呢? StrOuput 的结果 
pytorch 1.13.1 cuda11.6 Traceback (most recent call last): File "/EraseNet/train_STE.py", line 109, in G_loss.backward() File /.conda/envs/paddle_env/lib/python3.9/site-packages/torch/_tensor.py", line 488, in backward torch.autograd.backward( File ".conda/envs/paddle_env/lib/python3.9/site-packages/torch/autograd/__init__.py", line 197, in backward Variable._execution_engine.run_backward( # Calls...
你好,请问一下擦除效果会受到哪些因素影响,我使用你开源的模型进行文字擦除,效果不好,文本mask使用的文本检测模型,不知道是我代码问题,还是说受到训练数据的影响,导致我的预测结果不佳,图片的大小和非文本区域的数据分布会影响擦除效果吗,以下是我的原始图片ori和预测图片result  