File tree Expand file tree Collapse file tree 1 file changed +10
-4
lines changed
Expand file tree Collapse file tree 1 file changed +10
-4
lines changed Original file line number Diff line number Diff line change 2525#include < cstring>
2626
2727Monitor::SpeedAI::SpeedAI (Monitor *monitor_) :
28- monitor(monitor_)
28+ monitor(monitor_),
29+ module(nullptr )
2930{
3031}
3132
3233Monitor::SpeedAI::~SpeedAI () {
3334 // Clean up
34- uai_module_free (module );
35+ if (module )
36+ uai_module_free (module );
3537}
3638
3739bool 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
You can’t perform that action at this time.
0 commit comments