下面是一个使用 yolov3 在 PC 端处理视频流,并返回坐标给树莓派的代码示例:
``` import cv2 import numpy as np import time import socket
Load YOLO model
net = cv2.dnn.readNetFromDarknet("yolov3.cfg", "yolov3.weights") classes = [] with open("coco.names", "r") as f: classes = [line.strip() for line in f.readlines()] layer_names = net.getLayerNames() output_layers = [layer_names[i[0] - 1] for i in net.getUnconnectedOutLayers()] colors = np.random.uniform(0, 255, size=(len(classes), 3))
Start video capture
cap = cv2.VideoCapture(0)
Connect to Raspberry Pi
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM) client.connect(("192.168.1.100", 8000))
while Tru