Add keyboard exit catch

naisy · Dec 13, 2018 · 74da152 · 74da152
1 parent 4ea055f
commit 74da152
Show file tree

Hide file tree

Showing 7 changed files with 14 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -358,6 +358,8 @@ def start(self, cfg):
             """
             END while
             """
+        except KeyboardInterrupt:
+            pass
         except:
             import traceback
             traceback.print_exc()