You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
3.0 KiB
102 lines
3.0 KiB
import cv2
|
|
import socket
|
|
import json
|
|
import base64
|
|
from PIL import Image
|
|
from io import BytesIO
|
|
import psutil
|
|
import multiprocessing
|
|
cam = cv2.VideoCapture(0)
|
|
|
|
|
|
def open_switch():
|
|
|
|
HOST = '127.0.0.1' # Standard loopback interface address (localhost)
|
|
PORT = 8385 # Port to listen on (non-privileged ports are > 1023)
|
|
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
|
s.bind((HOST, PORT))
|
|
s.listen()
|
|
while 1:
|
|
conn, addr = s.accept()
|
|
with conn:
|
|
while True:
|
|
data = conn.recv(1024)
|
|
if not data:
|
|
break
|
|
conn.sendall(data)
|
|
|
|
img_counter = 0
|
|
|
|
encode_param = [int(cv2.IMWRITE_JPEG_QUALITY), 90]
|
|
|
|
socket_switch = True
|
|
cut=[-175,-1,-175,-1]
|
|
cut_send = [0,0,0,0]
|
|
data = {"gpu_temp":"10C","gpu_load":"15%","cpu_temp":"47C","cpu_load":"15%","mem_temp":"NaN","mem_load":"17%","fan_speed":"10000RPM"}
|
|
|
|
def get_temps():
|
|
global data
|
|
temps = psutil.sensors_temperatures()
|
|
data["cpu_temp"] = str(int(temps["dell_smm"][0][1]))+"°C"
|
|
data["cpu_load"] = str(psutil.cpu_percent())+"%"
|
|
data["mem_load"] = str(dict(psutil.virtual_memory()._asdict())["percent"])+"%"
|
|
data["fan_speed"] = str(psutil.sensors_fans()["dell_smm"][0][1])+"RPM"
|
|
|
|
p1 = multiprocessing.Process(target=open_switch)
|
|
p1.start()
|
|
|
|
while True:
|
|
try:
|
|
|
|
ret, frame = cam.read()
|
|
lens = [len(frame),0,len(frame[0])]
|
|
for i in range(0,len(cut),2):
|
|
if cut[i]<0:
|
|
cut_send[i] = lens[i] + cut[i]
|
|
cut_send[i+1] = abs(cut[i])-abs(cut[i+1])
|
|
backup = frame
|
|
frame = cv2.cvtColor(frame,cv2.COLOR_BGR2RGB)
|
|
crop_img = frame.copy(order='C')
|
|
|
|
crop_img = Image.fromarray(crop_img,"RGB")
|
|
buffered = BytesIO()
|
|
crop_img.save(buffered, format="JPEG")
|
|
img = base64.b64encode(buffered.getvalue()).decode("ascii")
|
|
frame_cut=backup[cut[0]:cut[1],cut[2]:cut[3]]
|
|
cv2.imshow("base",backup)
|
|
cv2.imshow("cut",frame_cut)
|
|
ex_c = [27, ord("q"), ord("Q")]
|
|
if cv2.waitKey(1) & 0xFF in ex_c:
|
|
break
|
|
if socket_switch:
|
|
try:
|
|
client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
client_socket.connect(('127.0.0.1', 8485))
|
|
connection = client_socket.makefile('wb')
|
|
socket_switch = False
|
|
except:
|
|
socket_switch=True
|
|
continue
|
|
try:
|
|
client_socket.sendall(json.dumps({"image_full":img,"image_sizes":{"x":cut_send[2],"y":cut_send[0],"width":cut_send[3],"height":cut_send[1]},"load":data}).encode('gbk')+b"\n")
|
|
print(img)
|
|
|
|
except:
|
|
socket_switch=True
|
|
img_counter += 1
|
|
if img_counter % 10 ==0:
|
|
get_temps()
|
|
except KeyboardInterrupt:
|
|
if not socket_switch:
|
|
client_socket.sendall(b"Bye\n")
|
|
cam.release()
|
|
p1.terminate()
|
|
break
|
|
|
|
cv2.destroyAllWindows()
|
|
p1.terminate()
|
|
|
|
|
|
|
|
|