From 74da1525e1b6bb0fae1955f07fb7bc655e238de5 Mon Sep 17 00:00:00 2001 From: Yoshiroh Takanashi <takanashi@gclue.jp> Date: Thu, 13 Dec 2018 10:20:46 +0900 Subject: [PATCH] Add keyboard exit catch --- lib/detection_deeplab_v3.py | 2 ++ lib/detection_faster_v2.py | 2 ++ lib/detection_mask_v1.py | 2 ++ lib/detection_nms_v0.py | 2 ++ lib/detection_nms_v1.py | 2 ++ lib/detection_trt_v1.py | 2 ++ lib/mtdetection_mask_v1.py | 2 ++ 7 files changed, 14 insertions(+) diff --git a/lib/detection_deeplab_v3.py b/lib/detection_deeplab_v3.py index 126e75f..7d15c43 100644 --- a/lib/detection_deeplab_v3.py +++ b/lib/detection_deeplab_v3.py @@ -395,6 +395,8 @@ def start(self, cfg): """ END while """ + except KeyboardInterrupt: + pass except: import traceback traceback.print_exc() diff --git a/lib/detection_faster_v2.py b/lib/detection_faster_v2.py index 008102d..bb4822b 100644 --- a/lib/detection_faster_v2.py +++ b/lib/detection_faster_v2.py @@ -422,6 +422,8 @@ def start(self, cfg): """ END while """ + except KeyboardInterrupt: + pass except: import traceback traceback.print_exc() diff --git a/lib/detection_mask_v1.py b/lib/detection_mask_v1.py index 6aed932..0094899 100644 --- a/lib/detection_mask_v1.py +++ b/lib/detection_mask_v1.py @@ -426,6 +426,8 @@ def start(self, cfg): """ END while """ + except KeyboardInterrupt: + pass except: import traceback traceback.print_exc() diff --git a/lib/detection_nms_v0.py b/lib/detection_nms_v0.py index 2c428aa..43c1114 100644 --- a/lib/detection_nms_v0.py +++ b/lib/detection_nms_v0.py @@ -390,6 +390,8 @@ def start(self, cfg): """ END while """ + except KeyboardInterrupt: + pass except: import traceback traceback.print_exc() diff --git a/lib/detection_nms_v1.py b/lib/detection_nms_v1.py index 2df32e7..f76d9d8 100644 --- a/lib/detection_nms_v1.py +++ b/lib/detection_nms_v1.py @@ -389,6 +389,8 @@ def start(self, cfg): """ END while """ + except KeyboardInterrupt: + pass except: import traceback traceback.print_exc() diff --git a/lib/detection_trt_v1.py b/lib/detection_trt_v1.py index 102d0aa..678fbb7 100644 --- a/lib/detection_trt_v1.py +++ b/lib/detection_trt_v1.py @@ -388,6 +388,8 @@ def start(self, cfg): """ END while """ + except KeyboardInterrupt: + pass except: import traceback traceback.print_exc() diff --git a/lib/mtdetection_mask_v1.py b/lib/mtdetection_mask_v1.py index 1c4612b..46f0b81 100644 --- a/lib/mtdetection_mask_v1.py +++ b/lib/mtdetection_mask_v1.py @@ -358,6 +358,8 @@ def start(self, cfg): """ END while """ + except KeyboardInterrupt: + pass except: import traceback traceback.print_exc()