PSENet.pytorch
PSENet.pytorch copied to clipboard
A pytorch re-implementation of PSENet: Shape Robust Text Detection with Progressive Scale Expansion Network
我原图是2000*3000,小目标文本像素在20*30的样子,训练好模型后进行检测,对于小目标文本存在漏检的情况,不知道作者有没有碰到
我想直接拿作者的predict函数来测试效果。但是在模型加载的时候出现模型key值不匹配的问题,怎么解决
希望大神空闲的话提供一下Windows版本哈
代码错误
感谢作者提供的优秀代码,在训练过程中我遇到一旦出现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 即可解决
你好,我在mac系统下的conda环境运行该程序,在这里报错。请问是怎么回事呢? 难道只能在linux下运行吗? 谢谢~
我在训练 时碰到 这个问题,你们遇到过吗?The kernel for PSENet/Untitled.ipynb appears to have died. It will restart automatically.