Skip to content

Commit c044d4e

Browse files
authored
Merge pull request #494 from Nuzhny007/master
Fixed MotionDetector example
2 parents fbab502 + ee879d2 commit c044d4e

File tree

2 files changed

+13
-4
lines changed

2 files changed

+13
-4
lines changed

example/MotionDetectorExample.h

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@ class MotionDetectorExample final : public VideoExample
2424
std::string bpeModel = "C:/work/clip/ruclip_/CLIP/data/ruclip-vit-large-patch14-336/bpe.model";
2525
m_clip.Init(clipModel, bpeModel, 336, 0, { "pedestrian", "person", "suv", "pickup", "car", "truck", "bus" });
2626
#endif // USE_CLIP
27+
28+
m_logger->info("MotionDetectorExample");
2729
}
2830

2931
protected:
@@ -34,7 +36,10 @@ class MotionDetectorExample final : public VideoExample
3436
///
3537
bool InitDetector(cv::UMat frame) override
3638
{
37-
m_minObjWidth = frame.cols / 20;
39+
m_logger->info("MotionDetectorExample::InitDetector");
40+
41+
//m_minObjWidth = frame.cols / 20;
42+
m_minObjWidth = 2;
3843

3944
config_t config;
4045
config.emplace("useRotatedRect", "0");
@@ -92,15 +97,17 @@ class MotionDetectorExample final : public VideoExample
9297
///
9398
bool InitTracker(cv::UMat frame) override
9499
{
100+
m_logger->info("MotionDetectorExample::InitTracker");
101+
95102
if (!m_trackerSettingsLoaded)
96103
{
97-
m_trackerSettings.SetDistance(tracking::DistRects);
104+
m_trackerSettings.SetDistance(tracking::DistCenters);
98105
m_trackerSettings.m_kalmanType = tracking::KalmanLinear;
99106
m_trackerSettings.m_filterGoal = tracking::FilterCenter;
100107
m_trackerSettings.m_lostTrackType = tracking::TrackNone; // Use visual objects tracker for collisions resolving. Used if m_filterGoal == tracking::FilterRect
101108
m_trackerSettings.m_matchType = tracking::MatchHungrian;
102109
m_trackerSettings.m_useAcceleration = false; // Use constant acceleration motion model
103-
m_trackerSettings.m_dt = m_trackerSettings.m_useAcceleration ? 0.05f : 0.5f; // Delta time for Kalman filter
110+
m_trackerSettings.m_dt = m_trackerSettings.m_useAcceleration ? 0.05f : 0.3f; // Delta time for Kalman filter
104111
m_trackerSettings.m_accelNoiseMag = 0.1f; // Accel noise magnitude for Kalman filter
105112
m_trackerSettings.m_distThres = 0.95f; // Distance threshold between region and object on two frames
106113
#if 1
@@ -138,7 +145,7 @@ class MotionDetectorExample final : public VideoExample
138145
///
139146
void DrawData(cv::Mat frame, const std::vector<TrackingObject>& tracks, int framesCounter, int currTime) override
140147
{
141-
m_logger->info("Frame ({1}): tracks = {2}, time = {3}", framesCounter, tracks.size(), currTime);
148+
m_logger->info("Frame ({0}): tracks = {1}, time = {2}", framesCounter, tracks.size(), currTime);
142149

143150
#ifdef USE_CLIP
144151
std::vector<CLIPResult> clipResult;

example/VideoExample.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,8 @@ void VideoExample::SyncProcess()
171171
m_logger->info("Process: riched last {} frame", m_endFrame);
172172
break;
173173
}
174+
175+
m_logger->debug("VideoExample::SyncProcess: Capture {0} frame", framesCounter);
174176
}
175177
if (i < m_batchSize)
176178
break;

0 commit comments

Comments
 (0)