-
Notifications
You must be signed in to change notification settings - Fork 0
/
client.py
78 lines (67 loc) · 2.04 KB
/
client.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
import socket
import cv2
import numpy
import os
import time
from picamera import PiCamera
from picamera.array import PiRGBArray
from RPLCD.gpio import CharLCD
import RPi.GPIO as GPIO
lcd = CharLCD(cols=16, rows=2, pin_rs=2, pin_rw=3, pin_e=4, pins_data=[5, 6, 7, 8], numbering_mode=GPIO.BCM)
GPIO.setwarnings(False)
#VIDEO_DIR = os.getcwd() + "/video/"
#IMAGE_DIR = os.getcwd() + "/img_output/"
#file_Name = VIDEO_DIR + "nano.mp4"
TCP_IP = '192.168.0.3'
TCP_PORT = 5001
print("Run")
#i=6
sock = socket.socket()
sock.connect((TCP_IP, TCP_PORT))
print("Socket create")
while(1):
camera = PiCamera()
camera.resolution = (1024, 512)
camera.resolution = (1024, 512)
camera.start_preview()
camera.color_effects = (128,128)
rawCapture = PiRGBArray(camera)
time.sleep(0.1)
camera.capture(rawCapture, format="bgr")
#img_gray = cv2.cvtColor(rawCapture, cv2.COLOR_BGR2GRAY)
image = rawCapture.array
print("image capture")
#capture = cv2.VideoCapture(file_Name)
#ret, image = capture.read()
#image = cv2.imread(IMAGE_DIR + "foo" + str(i) + ".jpg")
#r = 600.0 / image.shape[1]
#dim = (600, int(image.shape[0] * r))
# perform the actual resizing of the image and show it
#resized = cv2.resize(image, dim, interpolation = cv2.INTER_AREA)
encode_param=[int(cv2.IMWRITE_JPEG_QUALITY),90]
result, imgencode = cv2.imencode('.jpg', image, encode_param)
data = numpy.array(imgencode)
stringData = data.tostring()
sock.send( str(len(stringData)).ljust(16).encode())
sock.send( stringData )
str_msg = sock.recv(16).decode()
print(str_msg)
lcd.clear()
lcd.cursor_pos = (0, 0) #line1
lcd.write_string(str_msg)
lcd.cursor_pos = (0, 14)
lcd.write_string("%")
lcd.cursor_pos = (1, 3) #line2
lcd.write_string("confidence")
#i -= 1
#camera.stop_preview()
camera.close()
time.sleep(2)
#lcd.clear()
GPIO.setwarnings(False)
sock.close()
print("Socket close")
#decimg=cv2.imdecode(data,1)
#cv2.imshow('CLIENT',decimg)
#cv2.waitKey(0)
#cv2.destroyAllWindows()