search

lewes6369 / TensorRT-Yolov3

TensorRT for Yolov3
MIT License
489 stars 165 forks source link

AP is wrong #51

Closed sdyy closed 4 years ago

sdyy commented 5 years ago 

            int total = checkPRBoxs.size();  =>TP +FP

            for (const auto& item : checkPRBoxs)
            {
                item.second ? ++PR_TP : ++PR_FP;
                PRValues.emplace_back( make_pair(PR_TP/ float(PR_TP+PR_FP) , PR_TP / float(total)) );
            }

total should be TP + FN

lewes6369 commented 4 years ago

I am sorry to make the mistake. I fix it and eval the map again. Thanks for pointing out it.