handle mouse clicking window close icon to exit program

Author: swdee

cmd/yolo-detection/main.go

@@ -96,9 +96,14 @@ func main() {
 		detect(&net, &img, outputNames)
 
 		window.IMShow(img)
+
 		if window.WaitKey(1) >= 0 {
 			break
 		}
+		// break if user clicked the window close button
+		if window.GetWindowProperty(gocv.WindowPropertyVisible) < 1 {
+			break
+		}
 	}
 }