PSENet.pytorch

PSENet.pytorch copied to clipboard

针对小目标文本会有漏检的情况

2

我原图是2000*3000，小目标文本像素在20*30的样子，训练好模型后进行检测，对于小目标文本存在漏检的情况，不知道作者有没有碰到

Debugerss

predict.py测试出现问题

12

我想直接拿作者的predict函数来测试效果。但是在模型加载的时候出现模型key值不匹配的问题，怎么解决

Debugerss

希望大神空闲的话提供一下Windows版本哈

ouening

代码错误

3

感谢作者提供的优秀代码，在训练过程中我遇到一旦出现Loss小于0.4，运行一次测试集上的测试后，代码就会阻塞，无法继续训练。在对问题进行追踪后，我发现是DataLoader的写法有问题，导致在ubuntu上的多线程阻塞。解决方法： 1. config.py 将workers数量改为0，该方法会让数据读取变慢 2. train.py中的train_epoch里 for i, (images, labels, training_mask) in enumerate(train_loader): 改成 for i, mydatas in enumerate(train_loader): (images, labels, training_mask) = mydatas 即可解决

GuilmK

你好，我在mac系统下的conda环境运行该程序，在这里报错。请问是怎么回事呢？ 难道只能在linux下运行吗？ 谢谢～

ai1361720220000

我在训练 时碰到 这个问题，你们遇到过吗？The kernel for PSENet/Untitled.ipynb appears to have died. It will restart automatically.

zeng-cy