Fixed NMS seg fault #1064

This commit is contained in:
matlabbe
2023-06-20 20:53:32 -07:00
parent d901fb23e3
commit 337832e1b8

View File

@@ -2155,7 +2155,7 @@ void NMS(
{
for(int j = -dist_thresh; j < (dist_thresh+1); j++)
{
if(j==0 && k==0)
if((j==0 && k==0) || grid.at<unsigned char>(vv + k, uu + j) == 0)
continue;
if ( confidence.at<float>(vv + k - dist_thresh, uu + j - dist_thresh) <= c )