ValueError: cannot reshape array of size 7 into shape (2,2)

BADBADBADBOY / Psenet_v2

The code is based on resnet50. At present, the data set hmean in icdar2015 is about 80. A rough version will be sorted out and optimized later.

8 stars 3 forks source link

ValueError: cannot reshape array of size 7 into shape (2,2) #2

Open wuxiaolianggit opened 4 years ago

wuxiaolianggit commented 4 years ago

@BADBADBADBOY 您好，大佬，训练的时候出现这错误，您知道怎么解决吗？ File "/wuxl/Psenet_v2/metrics.py", line 19, in _fast_hist label_pred[mask], minlength=n_class**2).reshape(n_class, n_class) ValueError: cannot reshape array of size 7 into shape (2,2)

BADBADBADBOY commented 4 years ago

@wuxiaolianggit 看issues3

wuxiaolianggit commented 4 years ago

好的，多谢大佬的指导，我看一下PAN