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
@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)