Skip to content

Commit

Permalink
Add annotation of image
Browse files Browse the repository at this point in the history
  • Loading branch information
Isaac Connor committed Jan 25, 2025
1 parent fe4765a commit abfb07b
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions src/zm_netint_yolo.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
#include "zm_logger.h"
#include "zm_ffmpeg.h"
#include "zm_monitor.h"
#include "zm_vector2.h"

#include "zm_netint_yolo.h"

Expand Down Expand Up @@ -546,6 +547,8 @@ int Quadra_Yolo::process_roi(AVFrame *frame, AVFrame **filt_frame) {
Error("draw %d roi box failed", i);
return ret;
}
Image img(output);
img.Annotate(roi_class[roi_extra[i].cls], Vector2(roi[i].left, roi[i].top), monitor->LabelSize());

av_frame_free(&input);
input = output;
Expand Down

0 comments on commit abfb07b

Please sign in to comment.