| | |
| | | int onlyclass_id = 0; |
| | | for(t = 0; t < l.max_boxes; ++t){ |
| | | box truth = float_to_box(state.truth + t*5 + b*l.truths); |
| | | if(!truth.x) break; |
| | | if(!truth.x) break; // continue; |
| | | int class_id = state.truth[t*5 + b*l.truths + 4]; |
| | | float maxp = 0; |
| | | int maxi = 0; |
| | |
| | | box truth = float_to_box(state.truth + t*5 + b*l.truths); |
| | | int class_id = state.truth[t * 5 + b*l.truths + 4]; |
| | | if (class_id >= l.classes) continue; // if label contains class_id more than number of classes in the cfg-file |
| | | if(!truth.x) break; |
| | | if(!truth.x) break; // continue; |
| | | float iou = box_iou(pred, truth); |
| | | if (iou > best_iou) { |
| | | best_class_id = state.truth[t*5 + b*l.truths + 4]; |
| | |
| | | continue; // if label contains class_id more than number of classes in the cfg-file |
| | | } |
| | | |
| | | if(!truth.x) break; |
| | | if(!truth.x) break; // continue; |
| | | float best_iou = 0; |
| | | int best_index = 0; |
| | | int best_n = 0; |