Advertisement
Guest User

RPI

a guest
Nov 26th, 2014
148
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.85 KB | None | 0 0
  1. vid.py
  2. import numpy as np
  3. import cv2
  4.  
  5. cap = cv2.VideoCapture(0)
  6. cap.set(cv2.cv.CV_CAP_PROP_FRAME_WIDTH, 320)
  7. cap.set(cv2.cv.CV_CAP_PROP_FRAME_HEIGHT, 240)
  8.  
  9. while(True):
  10. # Capture frame-by-frame
  11. ret, frame = cap.read()
  12.  
  13. # Our operations on the frame come here
  14. gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
  15.  
  16. # Display the resulting frame
  17. cv2.imshow('frame',gray)
  18. if cv2.waitKey(1) & 0xFF == ord('q'):
  19. break
  20.  
  21. # When everything done, release the capture
  22. cap.release()
  23. cv2.destroyAllWindows()
  24.  
  25. udp.py
  26.  
  27. # python socket test
  28. import socket
  29. import rrb2 as rrb
  30.  
  31. rr = rrb.RRB2()
  32. host = '192.168.1.3'
  33. TCPport = 1523
  34.  
  35. def parse(rtxt):
  36. for delim in ' ':
  37. rtxt = rtxt.replace(delim, ' ')
  38. results = rtxt.split()
  39.  
  40. x = abs((float(results[0])-50)/50)
  41. y = abs((float(results[1])-50)/50)
  42. xr = ((float(results[0])-50)/50)
  43. yr = -((float(results[1])-50)/50)
  44. if(xr > 0):
  45. xd = 0
  46. else:
  47. xd = 1
  48. if(yr > 0):
  49. yd = 0
  50. else:
  51. yd = 1
  52. if(x > 0):
  53. x = 1
  54. else:
  55. x = 0
  56. if(y > 0):
  57. y = 1
  58. else:
  59. y = 0
  60.  
  61. #return results[0],results[0],results[1],results[1]
  62. #return abs((float(results[0])-50)/50),int((float(results[0])-50)/50),abs((float(results[1])-50)/50),int(-((float(results[1])-50)/50))
  63. #return ((float(results[0])-50)/50),int((float(results[0])-50)/50),((float(results[1])-50)/50),int(-((float(results[1])-50)/50))
  64. return x, xd, y, yd
  65.  
  66. s = socket.socket()
  67. s.connect((host,TCPport))
  68. while 1:
  69. raw = s.recv(128)
  70. #if raw == "LON":
  71. # rr.setOC1(1)
  72. #else:
  73. # if(raw == "LOFF"):
  74. # rr.setOC1(0)
  75. #else:
  76. # if(raw == "RON"):
  77. # rr.setOC2(1)
  78. #else:
  79. # if(raw == "ROFF"):
  80. # rr.setOC2(0)
  81. #else:
  82. spl, dirl, spr, dirr = parse(raw)
  83. print "x"
  84. print spl
  85. print "xd"
  86. print dirl
  87. print "y"
  88. print spr
  89. print "yd"
  90. print dirr
  91. rr.set_motors(spl, dirl, spr, dirr)
  92. s.close()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement