Merge pull request #11 from occ-ai/roy.fix_boxes_draw_class_name

refactor: Update draw_objects function to include classNames vector
locaal-ai · May 14, 2024 · b3000d9 · b3000d9
2 parents 1e1e865 + 8c1595d
commit b3000d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/detect-filter.cpp b/src/detect-filter.cpp
@@ -750,7 +750,7 @@ void detect_filter_video_tick(void *data, float seconds)
 
 	if (tf->preview || tf->maskingEnabled) {
 		if (tf->preview && objects.size() > 0) {
-			edgeyolo_cpp::utils::draw_objects(frame, objects);
+			edgeyolo_cpp::utils::draw_objects(frame, objects, tf->classNames);
 		}
 		if (tf->maskingEnabled) {
 			cv::Mat mask = cv::Mat::zeros(frame.size(), CV_8UC1);