update example for ilids

qbarthelemy · Nov 7, 2022 · 95334e2 · 95334e2
1 parent ef10080
commit 95334e2
Show file tree

Hide file tree

Showing 2 changed files with 76 additions and 41 deletions.
diff --git a/examples/dataset_ilids.py b/examples/dataset_ilids.py
@@ -4,7 +4,7 @@
 ===============================================================================
 
 1 - Download i-LIDS dataset;
-2 - Set paths and filenames;
+2 - Set dataset path and video name;
 3 - Run the script.
 """
 
@@ -15,46 +15,60 @@
 
 ###############################################################################
 
-# Dataset
-frame_filename = 'TODO' + '/Disk_1-Training/calibration/SZ2_far.tif'
-frame = cv.imread(frame_filename, cv.IMREAD_GRAYSCALE)
+# Dataset path
+data_path = r'TODO'
 
-is_view_1 = False
+# Video name
+video_name = 'SZTEA101a'
+
+# Switch between view1 and view2
+is_view_1 = True
+
+
+###############################################################################
 
 if is_view_1:
-    perimeter = np.array(
-        [[1.0       , 0.1145833  ],
-         [0.6       , 0.11458333],
-         [0.4625    , 0.30729167],
-         [0.014     , 0.640625],
-         [0.014     , 1.0       ],
-         [1.0       , 1.0       ]]
-    )
-    perspective = np.array(
-        [[0.65138889, 0.56770833],
-         [0.575     , 0.85243056],
-         [0.68333333, 0.20833333],
-         [0.72777778, 0.34548611]]
-    )
+    view = '1'
+    perimeter = np.array([
+        [1.0       , 0.1145833 ],
+        [0.6       , 0.11458333],
+        [0.4625    , 0.30729167],
+        [0.014     , 0.640625  ],
+        [0.014     , 1.0       ],
+        [1.0       , 1.0       ]
+    ])
+    perspective = np.array([
+        [0.65138889, 0.56770833],
+        [0.575     , 0.85243056],
+        [0.68333333, 0.20833333],
+        [0.72777778, 0.34548611]
+    ])
 else:
-    perimeter = np.array(
-        [[0.98611111, 0.69791667],
-         [0.62777778, 0.35590278],
-         [0.50555556, 0.17534722],
-         [0.50277778, 0.109375  ],
-         [0.43333333, 0.01388889],
-         [0.00416667, 0.00520833],
-         [0.00694444, 0.99826389],
-         [1.        , 1.        ]]
-    )
-    perspective = np.array(
-        [[0.49444444, 0.5625    ],
-         [0.40555556, 0.88541667],
-         [0.31388889, 0.06944444],
-         [0.275     , 0.19965278]]
-    )
-
-# Pipeline GOFPID
+    view = '2'
+    perimeter = np.array([
+        [0.98611111, 0.69791667],
+        [0.62777778, 0.35590278],
+        [0.50555556, 0.17534722],
+        [0.50277778, 0.109375  ],
+        [0.43333333, 0.01388889],
+        [0.00416667, 0.00520833],
+        [0.00694444, 0.99826389],
+        [1.        , 1.        ]
+    ])
+    perspective = np.array([
+        [0.49444444, 0.5625    ],
+        [0.40555556, 0.88541667],
+        [0.31388889, 0.06944444],
+        [0.275     , 0.19965278]
+    ])
+
+
+# Intrusion detection config
+
+config_frame_filename = (
+    data_path + '/Disk_1-Training/calibration/SZ' + view + '_far.tif'
+)
+
 gofpid = GOFPID(
     post_filter={
         'display_config': True,
@@ -64,12 +78,33 @@
         'perspective_coeff': 0.5,
         'presence_min': 3,
         'distance_min': 0.25,
-        'config_frame': frame,
+        'config_frame': cv.imread(config_frame_filename, cv.IMREAD_GRAYSCALE),
     },
     verbose=True
 ).initialize()
 
-# Intrusion detection
-# TODO
+
+# Intrusion detection test
+
+video_filename = data_path + '/Disk_2-Testing/video/' + video_name + '.mov'
+vidcap = cv.VideoCapture(video_filename)
+if not vidcap.isOpened():
+    print('Unable to open video file :', video_filename)
+    exit(0)
+
+while True:
+    _, frame = vidcap.read()
+    if frame is None:
+        break
+
+    y = gofpid.detect(frame)
+    gofpid.display(frame)
+    cv.imshow('Frame', frame)
+
+    if cv.waitKey(1) & 0xFF == ord('c'):
+        vidcap.release()
+        cv.destroyAllWindows()
+        break
+
 
 ###############################################################################
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pygofpid"
-version = "0.3.1"
+version = "0.3.2"
 description = "PyGOFPID is a Python package for good old fashioned perimeter intrusion detection systems"
 authors = ["Quentin Barthelemy"]
 license = "BSD-3-Clause"