added check to detectNet::Overlay()

kelo-robotics · Jul 28, 2020 · e7d6f91 · e7d6f91
1 parent 8b399c1
commit e7d6f91
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/c/detectNet.cpp b/c/detectNet.cpp
@@ -1062,6 +1062,13 @@ bool detectNet::Overlay( void* input, void* output, uint32_t width, uint32_t hei
 		}
 	}
 
+	// make sure there are actually detections
+	if( numDetections <= 0 )
+	{
+		PROFILER_END(PROFILER_VISUALIZE);
+		return true;
+	}
+
 	// bounding box overlay
 	if( flags & OVERLAY_BOX )
 	{