Skip to content

Commit 07e2489

Browse files
Initialize module to null, add some logging change location of model
1 parent d20914f commit 07e2489

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

src/zm_monitor_speedai.cpp

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,20 +25,26 @@
2525
#include <cstring>
2626

2727
Monitor::SpeedAI::SpeedAI(Monitor *monitor_) :
28-
monitor(monitor_)
28+
monitor(monitor_),
29+
module(nullptr)
2930
{
3031
}
3132

3233
Monitor::SpeedAI::~SpeedAI() {
3334
// Clean up
34-
uai_module_free(module);
35+
if (module)
36+
uai_module_free(module);
3537
}
3638

3739
bool Monitor::SpeedAI::setup() {
3840
// Load and launch module
39-
std::string model = "/usr/share/zoneminder/u_speedai_yolo_model_creator.uxf";
41+
std::string model = "/var/cache/zoneminder/models/u_speedai_yolo_model_creator.uxf";
4042
Debug(1, "SpeedAI: Loading model %s", model.c_str());
41-
uai_module_load(model.c_str(), &module);
43+
UaiErr err = uai_module_load(model.c_str(), &module);
44+
if (err != UAI_SUCCESS) {
45+
Error("Failed loading model %s", uai_err_string(err));
46+
return false;
47+
}
4248
Debug(1, "SpeedAI: launching");
4349
uai_module_launch(module);
4450
// Get info on input/output streams. We assume a simple model (like Resnet50) with one input

0 commit comments

Comments
 (0)