Efficient and Accurate Arbitrary-Shaped Text Detection with Pixel Aggregation Network模型实现

由于排版方便采用图片形式,具体参考文章如下,已给出链接。

本文为2019年文本检测期刊上的Efficient and Accurate Arbitrary-Shaped Text Detection with Pixel Aggregation Network的论文复现。

 

 如果出现cuda:0的话;说明你的机器模型中gpu问题,如果只是调用predict.py的文件的话,可以采用预训练模型,然后强制使用cpu跑模型。

具体修改的地方就在predict.py文件中加载模型的那个类里面。改为self.device('cpu')

猜你喜欢

转载自www.cnblogs.com/windfish/p/12056830.html
今日推荐