/
breakVideoToFrames.py
93 lines (70 loc) · 2.34 KB
/
breakVideoToFrames.py
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
# Working directory: "Face Tracker" (not needed for colab)
import sys
sys.path.append("D:/Python/Face Tracker")
import os
# Necessary Packages
import numpy as np
import time
import cv2
# Necessary Repos
# Necessary Files
import drawframe
import organizefiles
from facematcher import Matching
def breakdown(inputFileFolder, input_short, input_name):
# Input:
# Open MP4 Input
cap = organizefiles.openInputVideo(inputFileFolder, input_name)
videoFolderName = "D:/Python/Face Tracker/VideosFrameByFrame" + "/" + input_short
os.mkdir(videoFolderName)
for frame_num in range(int(cap.get(cv2.CAP_PROP_FRAME_COUNT))):
ret, frame = cap.read()
if ret:
frameFileName = videoFolderName + "/" + str(frame_num) + ".jpg"
cv2.imwrite(filename = frameFileName, img = frame)
else:
break
# Close MP4 Input
cap.release()
def generateWiderValTxt():
#Open TXT Output
out = open("D:/Python/Face Tracker/wider_val.txt", "w")
'''
for i in range(478):
out.write("/0--onemantwowoman/{}.jpg".format(str(i)) + '\n')
for i in range(391):
out.write("/5--fourhallway/{}.jpg".format(str(i)) + '\n')
for i in range(705):
out.write("/6--bigcrowd/{}.jpg".format(str(i)) + '\n')
for i in range(821):
out.write("/7--rainpedestrians/{}.jpg".format(str(i)) + '\n')
'''
for i in range(1525):
out.write("/8--voccamp/{}.jpg".format(str(i)) + '\n')
out.close()
inputFileFolder = "sourceVideos"
input_videos = {
"voccamp" : "voccamp",
}
'''
Inactive Videos:
Have been run before, used with up-to-date retinaface:
"rainpedestrians" : "crowdedstreetundertherain-pexels",
"bigcrowd" : "skywalkmahanakhon-videvo",
"fourhallway" : "walkinghallway-pexels",
"onemantwowoman" : "onemantwowomen_face-demographics-walking-and-pause",
Have never been run yet:
"panning" : "dogrunning",
"oneman" : "oneman_face-demographics-walking-and-pause",
"onewoman" : "onewoman_face-demographics-walking-and-pause",
"onemanonewoman" : "onemanonewoman_face-demographics-walking-and-pause",
'''
'''
for input_short, input_name in input_videos.items():
breakdown(
inputFileFolder = inputFileFolder,
input_short = input_short,
input_name = input_name,
)
'''
generateWiderValTxt()