基于python的移动物体检测与跟踪算法_详解使用PyTorch实现目标检测与跟踪

python教程栏目介绍使用PyTorch实现目标检测与跟踪

大量免费学习推荐，敬请访问python教程(视频)

引言

在昨天的文章中，我们介绍了如何在PyTorch中使用您自己的图像来训练图像分类器，然后使用它来进行图像识别。本文将展示如何使用预训练的分类器检测图像中的多个对象，并在视频中跟踪它们。

图像中的目标检测

目标检测的算法有很多，YOLO跟SSD是现下最流行的算法。在本文中，我们将使用YOLOv3。在这里我们不会详细讨论YOLO，如果想对它有更多了解，可以参考下面的链接哦~(https://pjreddie.com/darknet/yolo/)

下面让我们开始吧，依然从导入模块开始：from models import *

from utils import *

import os, sys, time, datetime, random

import torch

from torch.utils.data import DataLoader

from torchvision import datasets, transforms

from torch.autograd import Variable

import matplotlib.pyplot as plt

import matplotlib.patches as patches

from PIL import Image

然后加载预训练的配置和权重，以及一些预定义的值，包括：图像的尺寸、置信度阈值和非最大抑制阈值。config_path='config/yolov3.cfg'

weights_path='config/yolov3.weights'

class_path='config/coco.names'

img_size=416

conf_thres=0.8

nms_thres=0.4

# Load model and weights

model = Darknet(config_path, img_size=img_size)

model.load_weights(weights_path)

model.cuda()

model.eval()

classes = utils.load_classes(class_path)

Tensor = torch.cuda.FloatTensor

下面的函数将返回对指定图像的检测结果。def detect_image(img):