def process_img(frame):
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
img = imgprocess.imageDW(gray,(common_config.CAP_HEIGHT,common_config.CAP_WIDTH),1)
new_img = np.zeros(img.shape)
for i in range(img.shape[0]):
for j in range(img.shape[1]):
new_img[i][j] = 0 if img[i][j] < THRESHOLD else 255
new_img = new_img.ravel()
new_img = [y/255.0 for y in new_img]
new_img = np.reshape(new_img, (common_config.NETWORK_INPUT_SIZE, 1))
return new_img
cv2.imwrite(name, image)
## Main function
# major variables
cap = cv2.VideoCapture(0)
mo = move.Move()
while(cap.isOpened()):
# Capture frame-by-frame
ret, frame = cap.read()
# Our operations on the frame come here
# Change frame to gray level image
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
img = imgprocess.imageDW(gray,gray.shape,4)
# Display the resulting frame
cv2.imshow('frame',frame)
# Choose direction or quit
input_key = cv2.waitKey(1) & 0xFF
if input_key == ord('q'):
break
elif input_key == ord('w'):
mo.forward(common_config.SLEEP_TIME)
save_image("w", gray)
elif input_key == ord('s'):
mo.backward(common_config.SLEEP_TIME)
save_image("s", gray)
elif input_key == ord('a'):
print "want to save file:"+name
cv2.imwrite(name, image)
## Main function
# major variables
cap = cv2.VideoCapture(1)
while(cap.isOpened()):
# Capture frame-by-frame
ret, frame = cap.read()
# Our operations on the frame come here
# Change frame to gray level image
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
img = imgprocess.imageDW(gray,gray.shape,2)
# Do not display the resulting frame
cv2.imshow('frame',img)
# Choose direction or quit
input_key = cv2.waitKey(1) & 0xFF
if input_key == ord('q'):
break
elif input_key == ord('w'):
save_image("w", img)
elif input_key == ord('s'):
save_image("s", img)
elif input_key == ord('a'):
save_image("a", img)
elif input_key == ord('d'):
tr = GetFrameThread()
tr.start()
## main loop
pre_a = np.ndarray((3, 1))
while (cap.isOpened()):
# Capture frame-by-frame
# ret, frame = cap.read()
ret, current_frame = getFrame()
# Our operations on the frame come here
# Change frame to gray level image and do some trasition
gray = cv2.cvtColor(current_frame, cv2.COLOR_BGR2GRAY)
img = imgprocess.imageDW(
gray, (common_config.CAP_HEIGHT, common_config.CAP_WIDTH), 1)
new_img = np.zeros(img.shape)
for i in range(img.shape[0]):
for j in range(img.shape[1]):
new_img[i][j] = 0 if img[i][j] < THRESHOLD else 255
new_img = new_img.ravel()
new_img = [y / 255.0 for y in new_img]
new_img = np.reshape(new_img, (common_config.NETWORK_INPUT_SIZE, 1))
# decide direction
a = net.feedforward(new_img)
print a
if (pre_a == a).all():
print "Image not change. ignore this turn"
continue
pre_a = a.copy()
return ret, frame
tr = GetFrameThread()
tr.start()
## main loop
pre_a = np.ndarray((3,1))
while(cap.isOpened()):
# Capture frame-by-frame
# ret, frame = cap.read()
ret, current_frame = getFrame()
# Our operations on the frame come here
# Change frame to gray level image and do some trasition
gray = cv2.cvtColor(current_frame, cv2.COLOR_BGR2GRAY)
img = imgprocess.imageDW(gray,(common_config.CAP_HEIGHT,common_config.CAP_WIDTH),1)
new_img = np.zeros(img.shape)
for i in range(img.shape[0]):
for j in range(img.shape[1]):
new_img[i][j] = 0 if img[i][j] < THRESHOLD else 255
new_img = new_img.ravel()
new_img = [y/255.0 for y in new_img]
new_img = np.reshape(new_img, (common_config.NETWORK_INPUT_SIZE, 1))
# decide direction
a = net.feedforward(new_img)
print a
if (pre_a == a).all():
print "Image not change. ignore this turn"
continue
pre_a = a.copy()
