Automated-Proctoring/plot_graphs.py at main · Atsurak/Automated-Proctoring · 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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
import numpy as np
import time
import matplotlib.pyplot as plt
from testing import *

def print_list(items):
    for item in items:
        print(item,end=',')
    print()

def segment_list(items,segment_time,fps_assumed):
    no_of_frames = segment_time*fps_assumed
    count = 0
    listcount = []
    curr = 0
    for item in items:
        curr+=1
        if item == 1:
            count+=1

        if curr%no_of_frames == 0:
            listcount.append((count/no_of_frames)*100)
            count=0
    print_list(listcount)

def plot_main(frames,segment_time,fps_assumed):
    noface=[]
    multiface=[]
    facerec=[]
    head=[]
    mouth=[]
    spoof=[]
    cheat=[]
    seg=[]
    no_of_frames = segment_time*fps_assumed
    for frame in frames:
        noface.append(frame.noface)
        multiface.append(frame.multiface)
        facerec.append(frame.facerec)
        head.append(frame.head)
        mouth.append(frame.mouth)
        spoof.append(frame.spoof)
        cheat.append(frame.cheat)
        if(len(noface)%no_of_frames==0):
            seg.append(len(noface))
    cheat_list = [noface, multiface, facerec, head, mouth, spoof, cheat]
    titles_list = ["No face Detected", "Multiple faces Detected", "User Authentication", "Looking away from screen", "Speaking", "Spoofing Detected", "Estimated cheat Status"]

    plt.figure(figsize=(12,20))
    for i in range(7):
        plt.subplot(7,1,i+1)
        plt.step(cheat_list[i],'r')
        plt.yticks([0,1])
        #plt.xticks(seg)
        plt.title(titles_list[i])
        if(i==3):
            plt.ylabel("cheat status")
    plt.xlabel("frames")
    plt.subplots_adjust(hspace = 1.0)
    plt.savefig("results/cheat_frames_"+ time.strftime("%Y%m%d-%H%M%S") + ".png",dpi=300)
    plt.show()

    print("Segment wise results")
    for i in range(len(cheat_list)):
        print(titles_list[i])
        segment_list(cheat_list[i],segment_time,fps_assumed)

def cheat_count(segments):
    cheat_count =0
    for segment in segments:
        if(segment.cheat):
            cheat_count+=1
    return cheat_count

def plot_segments(segments,segment_time,original=[]):
    x=[]
    detected=[]
    n = cheat_count(segments)
    for segment in segments:
        x.append(segment.count)
        detected.append(segment.cheat)

    plt.figure(figsize=(12,4))
    plt.step(detected,'r')
    plt.step(original,'b')
    plt.yticks([0,1])
    #plt.xticks(x)
    plt.xlabel('Time Segments')
    plt.ylabel('Cheating Suspected')
    stats = "Total Time : " + str(len(segments)*segment_time) + " seconds\n" + "Cheating Suspected for : " + str(n*segment_time) + " seconds"
    plt.figtext(0.5, 0.9,stats, ha="center", fontsize=12, bbox={"facecolor":"orange", "alpha":0.5, "pad":3})
    plt.savefig("results/cheating_detection_"+ time.strftime("%Y%m%d-%H%M%S") + ".png",dpi=300)
    plt.show()
    print(x)
    print(detected)