【发疯毕设日志day7】大疆tello火影结印起飞——模型测试代码

叫我Python小朋友

于 2024-03-23 21:25:33 发布

阅读量363

点赞数 4

文章标签：目标跟踪人工智能计算机视觉

本文链接：https://blog.youkuaiyun.com/pythonAk47vt/article/details/136597238

版权

1. 写了一个类 yolo_test

启动tello，加载初始化模型

设置一个置信度阈值（设置的高一些）

import cv2
import numpy as np
from djitellopy import Tello

import torch

class yolo_test:
    def __init__(self):

        # tello初始化
        self.tello = Tello()
        self.tello.connect()
        self.tello.streamon()
        # 加载 yolov5模型
        self.model = torch.hub.load('./yolov5', 'custom', path='./weights/best.pt',source='local')
        # 置信度阈值
        self.model.conf = 0.7

2. 读取视频流转化成RGB

解析目标检测结果

3.

"""
测试训练号的YOLO模型
"""

import cv2
import numpy as np
from djitellopy import Tello

import torch

class yolo_test:
    def __init__(self):

        # tello初始化
        self.tello = Tello()
        self.tello.connect()
        self.tello