Merge pull request #61 from echo1consulting/master-keyboard-catch

Catch keyboard exit
naisy · Dec 12, 2018 · 2eda8a4 · 2eda8a4
2 parents 8297770 + 94f8f90
commit 2eda8a4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/lib/detection_nms_v2.py b/lib/detection_nms_v2.py
@@ -390,6 +390,8 @@ def start(self, cfg):
             """
             END while
             """
+        except KeyboardInterrupt:
+            pass
         except:
             import traceback
             traceback.print_exc()

diff --git a/lib/mpfps.py b/lib/mpfps.py
@@ -186,6 +186,8 @@ def process_fps_counter(self):
                         MPVariable.vis_fps.value = 0
                         MPVariable.vis_skip_rate.value = 0
                     previos_work_time = now_time
+        except KeyboardInterrupt:
+            pass
         except Exception as e:
             import traceback
             traceback.print_exc()
@@ -258,6 +260,8 @@ def process_fps_console(self):
                     MPVariable.vis_proc_time.value = 0
                     MPVariable.vis_drop_frames.value = 0
                     previos_work_time = now_time
+        except KeyboardInterrupt:
+            pass
         except Exception as e:
             import traceback
             traceback.print_exc()