Backend-FastAPI/displacement_detection.py at main · TeamRenegan/Backend-FastAPI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import cv2

def detect_cctv_fall(image1, image2, threshold=1000, edges_threshold=0.10, debug=False):
    img1_gray = cv2.cvtColor(image1, cv2.COLOR_BGR2GRAY)
    img2_gray = cv2.cvtColor(image2, cv2.COLOR_BGR2GRAY)

    img1_gaussian = cv2.GaussianBlur(img1_gray, (5, 5), 0)
    img2_gaussian = cv2.GaussianBlur(img2_gray, (5, 5), 0)

    img1_edges = cv2.Canny(img1_gaussian, 100, 300)
    img2_edges = cv2.Canny(img2_gaussian, 100, 300)

    img1_edges_var = img1_edges.var()
    img2_edges_var = img2_edges.var()

    subtracted = cv2.subtract(img1_gaussian, img2_gaussian)
    variation = subtracted.var()

    if debug:
        print("Variation: ", variation)
        print("Edges Variation: ", img1_edges_var, img2_edges_var)
        print("Edges Variation Difference: ", abs(img1_edges_var - img2_edges_var))

        while True:
            cv2.imshow("Image1", image1)
            cv2.imshow("Image2", image2)
            cv2.imshow("Edges1", img1_edges)
            cv2.imshow("Edges2", img2_edges)
            cv2.imshow("Gaussion1", img1_gaussian)
            cv2.imshow("Gaussion2", img2_gaussian)
            cv2.imshow("Subtracted", subtracted)

            if cv2.waitKey(1) == ord("q"):
                break

    if (
        variation > threshold
        or abs(img1_edges_var - img2_edges_var) / img1_edges_var > edges_threshold
    ):
        return True
    else:
        return False