def goGetEm():
minRemovalScore = 0.25
timeOut = 10
cleanThresh = 5
binNumber = 100
distanceWeight = 0.5
timeWeight = 0.0
sizeWeight = 0.5
weights = (distanceWeight, timeWeight, sizeWeight)
writingToFiles = True
distDev = 200
timeDev = 1
sizeDev = 0.25
devs = (distDev, timeDev, sizeDev)
framesback = 5
# minRemovals = [0.0001, 0.1, 0.25]
# weightDev = [((0.5,0,0.5),(100,0.34,0.15)),((0.5,0,0.5),(200,0.34,0.25)),((0.45,0.2,0.35),(100,0.34,0.15)),((0.45,0.2,0.35),(200,0.34,0.25))]
i = 0
# for minRemovalScore in minRemovals:
# for setting in weightDev:
# weights = setting[0]
# devs = setting[1]
variables = [minRemovalScore, timeOut, cleanThresh, binNumber, weights, writingToFiles, devs, framesback]
vidFile = "datums/outvideo"+str(i)+".avi"
vid = Video("Testvideos/Movie1.mov", variables, False)
vid.openVidWrite(vidFile)
success = vid.readFrame()
while (success):
vid.findFaces()
vid.writeToVideo()
success = vid.readFrame()
vid.endWindow()
def goGetEm(): minRemovalScore = 0.0001 timeOut = 10 cleanThresh = 5 binNumber = 100 distanceWeight = 0.5 sizeWeight = 0.0 timeWeight = 0.5 weights = (distanceWeight, timeWeight, sizeWeight) writingToFiles = True distDev = 200 timeDev = 0.34 sizeDev = 0.25 devs = (distDev, timeDev, sizeDev) framesback = 2 variables = [minRemovalScore, timeOut, cleanThresh, binNumber, weights, writingToFiles, devs, framesback] vid = Video(0, variables) # vidFile = "outvid.avi" # csvFile = "variable.csv" # if writingToFiles: # vid.openVidWrite(vidFile) # vid.openCSVWrite(csvFile) while (True): vid.readFrame() vid.findFaces() vid.display() # vid.writeToVideo() # exit on escape key key = cv2.waitKey(1) if key == 27: break vid.endWindow()
def goGetEm():
minRemovalScore = 0.0001
timeOut = 10
cleanThresh = 5
binNumber = 100
distanceWeight = 1
timeWeight = 1
sizeWeight = 1
#weights = (distanceWeight, timeWeight, sizeWeight)
#variables = [minRemovalScore, timeOut, cleanThresh, binNumber, weights]
variables = (0.25, 10, 5, 100, (1,1,1), False, (200,0.34,0.25), 2)
#vid = Video("Slightmovement.MP4", [0,1,0.1,0.1,1000,0])
vid = Video(0, variables)
while (True):
vid.readFrame()
vid.findFaces()
vid.display()
# exit on escape key
key = cv2.waitKey(20)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
vid.endWindow()
def draw(self):
'''Draws main GUI, then loops on frames'''
grid = QGridLayout()
self.leftbox = ImageBox()
grid.addWidget(self.leftbox,0,0)
self.rightbox = ControlBox(self)
grid.addWidget(self.rightbox, 0, 1)
variables = (0.25, 10, 5, 100, (0.5,0,0.5), False, (200.0,0.34,0.25), 2)
#create Video object
vid = Video(0,variables)
while(True):
# Read every frame and extract images
vid.readFrame()
frame_as_string_before = vid.getCurrentFrame().tostring()
vid.findFaces()
self.face_list = vid.getFaces()
# If we've clicked the num button, draw rectangles around each face
if self.draw_nums:
for i in range(len(self.face_list)):
# Uncomment if we don't want to use predicted position
# vid.showRectangle(self.face_list[i].getPosition(),self.face_list[i].getID())
# If the face is obscured, draw the rectangle around the predicted position
if not face.isObscured():
vid.showRectangle(self.face_list[i].getPosition(),self.face_list[i].getID())
else:
vid.showRectangle(self.face_list[i].getPredictedPosition(),self.face_list[i].getID())
frame = vid.getCurrentFrame()
rects = []
#Get position of each face
for face in self.face_list:
# Position of each face (for use in Control Box) based on last detected position, not predicted
tuples = face.getPosition()
rects.append([tuples[0][0], tuples[0][1], tuples[1][0], tuples[1][1]])
# Uncomment if we want to base it on predicted
# if not face.isObscured():
# tuples = face.getPosition()
# rects.append([tuples[0][0], tuples[0][1], tuples[1][0], tuples[1][1]])
# else:
# tuples = face.getPredictedPosition()
# if tuples:
# rects.append([tuples[0][0], tuples[0][1], tuples[1][0], tuples[1][1]])
# Transform cv2 frame (numpy array) into QPixmap via string and QImage
cv2.cvtColor(frame, cv.CV_BGR2RGB, frame)
frame_as_string = frame.tostring()
image = QImage(frame_as_string,\
frame.shape[1],frame.shape[0],QImage.Format_RGB888)
pixmap = QPixmap.fromImage(image)
# Get images of faces for use in Control Box
face_pics = get_imgs_from_rects(image, rects)
# DON'T DELETE
# What does it do
# WHO KNOWS
if cv2.waitKey(1) & 0xFF == ord('q'):
vid.endWindow()
break
# Update everything with current face info
self.leftbox.set_image(pixmap, self.face_list, face_pics)
self.rightbox.setFaces(face_pics, self.face_list)
self.setLayout(grid)
self.show()