VCC/vcs-python-repository
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

updated to use jpeg decoding

Browse source
This commit is contained in:
Nickiel12 2024-06-22 12:53:45 -07:00
parent 4fa94a647d
commit d7d5887fb1
2 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
Server/WebsocketServer/main.py
View file

@ -2,6 +2,7 @@ import websockets
import asyncio import asyncio
import numpy as np import numpy as np
from ultralytics import YOLO from ultralytics import YOLO
import time
import cv2 import cv2
classNames = ["person", "bicycle", "car", "motorbike", "aeroplane", "bus", "train", "truck", "boat", classNames = ["person", "bicycle", "car", "motorbike", "aeroplane", "bus", "train", "truck", "boat",
@ -28,6 +29,7 @@ async def handle_connection(websocket, path):
# frame = cv2.imdecode(nparr, cv2.IMREAD_COLOR) # frame = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
frame = cv2.cvtColor(nparr, cv2.COLOR_BGR2RGB) frame = cv2.cvtColor(nparr, cv2.COLOR_BGR2RGB)
cv2.imshow("from_remote", frame)
# Perform object detection # Perform object detection
results = model.track(frame, persist=True) results = model.track(frame, persist=True)
@ -73,6 +75,8 @@ async def handle_connection(websocket, path):
ret = lines.encode('utf-8') ret = lines.encode('utf-8')
cv2.waitKey(80)
await websocket.send(ret) await websocket.send(ret)
except websockets.exceptions.ConnectionClosed: except websockets.exceptions.ConnectionClosed:

2
joystick-controller-client

@ -1 +1 @@
Subproject commit 66acd4698fcc997099ebc1ddc6f8e5ddd0e3579c Subproject commit a3d8a6fd064c0a194dc0815e7f19c900937921d9