-
Notifications
You must be signed in to change notification settings - Fork 249
Closed
Description
Hi,
It seems to me that the line
pred[w,h,int(label[w,h,1,n])+1,n] >= maximum(pred[w, h, 1:channels, n])
contains a potential bug if the prediction scores are, for instance, identically zero.
Note that this is very unlikely to happen after training starts, but it might be nice to fix just in case.
It might be better to compute an argmax of pred and then check that it is equal to the correct label.
-Nick
Metadata
Metadata
Assignees
Labels
No labels