你好，
在使用CRAFT从头开始训练SynthText数据集时，我不知道为什么会出现尺寸不一致的问题？
训练脚本是：CUDA_VISIBLE_DEVICES=0 python3 trainSynth.py --yaml=syn_train
以下是相关信息：
2023-02-08 01:38:12, training_step: 49950|50000, learning rate: 0.00000080, training_loss: 0.03255, avg_batch_time: 29187.57935
2023-02-08 01:38:14, training_step: 49955|50000, learning rate: 0.00000080, training_loss: 0.03671, avg_batch_time: 29189.06155
错误：操作数的形状(200,200)与(23,0)无法广播在一起。
在生成亲和力图时，出现了一个奇怪的框(宽度： 0,高度： 23)。
我不知道是否有人能帮助解决这个问题。谢谢。
4条答案
按热度按时间9lowa7mx1#
我也有同样的问题，如果有人发现了这个问题，请告诉我。
gmol16392#
你好，我不知道在使用从头开始训练CRAFT的SynthText数据集时，尺寸不一致是如何发生的？训练脚本是：CUDA_VISIBLE_DEVICES=0 python3 trainSynth.py --yaml=syn_train
相关信息如下：2023-02-08 01:38:12, training_step: 49950|50000, learning rate: 0.00000080, training_loss: 0.03255, avg_batch_time: 29187.57935 2023-02-08 01:38:14, training_step: 49955|50000, learning rate: 0.00000080, training_loss: 0.03671, avg_batch_time: 29189.06155 错误：操作数无法广播到形状 (200,200) (23,0) 在生成亲和力图时，出现了奇怪的框(宽度： 0,高度： 23)。
我不知道是否有人能帮助解决这个问题。谢谢。
你知道权重文件存储在哪里吗？
ecfdbz9o3#
首先，这种错误的原因都是生成亲和图和框生成错误。具体错误是，在处理模型得到的Map后，得到了两个框，这两个框就是预测的两个字符。经过这两个框后，生成了一个框，这个框是用来生成亲和图的两个字符中间的部分。在这个点上，如果生成的框是单列的，但是计算是基于一行的，那么它的宽度就会为零。
ctehm74n4#
首先，这种错误的原因都是生成亲和力图和框生成错误。具体的错误是，在处理模型得到的Map后，得到了两个框，这两个框分别是预测的两个字符。经过这两个框后，生成了一个框，这个框是用来生成亲和力图的两个字符中间的部分。在这个点上，如果生成的框是单列的，但是计算是基于一行的，那么它的宽度就会为零。
如何解决？