-
Notifications
You must be signed in to change notification settings - Fork 0
/
imageCollector.py
42 lines (33 loc) · 1.12 KB
/
imageCollector.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
import os
import uuid
import cv2
# os.path.join(ANC_PATH, '{}.jpg'.format(uuid.uuid1()))
POS_PATH = os.path.join("data", "positive")
NEG_PATH = os.path.join("data", "negative")
ANC_PATH = os.path.join("data", "anchor")
cap = cv2.VideoCapture(0)
while cap.isOpened():
ret, frame = cap.read()
# Cut down frame to 250x250px
frame = frame[120:120+250,200:200+250, :]
# Collect anchors
if cv2.waitKey(1) & 0XFF == ord('a'):
# Create the unique file path
imgname = os.path.join(ANC_PATH, '{}.jpg'.format(uuid.uuid1()))
# Write out anchor image
cv2.imwrite(imgname, frame)
# Collect positives
if cv2.waitKey(1) & 0XFF == ord('p'):
# Create the unique file path
imgname = os.path.join(POS_PATH, '{}.jpg'.format(uuid.uuid1()))
# Write out positive image
cv2.imwrite(imgname, frame)
# Show image back to screen
cv2.imshow('Image Collection', frame)
# Breaking gracefully
if cv2.waitKey(1) & 0XFF == ord('q'):
break
# Release the webcam
cap.release()
# Close the image show frame
cv2.destroyAllWindows()