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()