利用pycharm阅读代码,进行Debug
objdetector.py 注释
import torch
import numpy as np
from models.experimental import attempt_load
from utils.general import non_max_suppression, scale_coords
from utils.datasets import letterbox
from utils.torch_utils import select_device
import objtracker
# 要检测的类别,这里只检测人,车(小车、巴士、卡车)
OBJ_LIST = ['person', 'car', 'bus', 'truck']
# yolov5模型
DETECTOR_PATH = 'weights/yolov5m.pt'
class baseDet(object):
def __init__(self):
self.img_size = 640
self.threshold = 0.3
self.stride = 1
def build_config(self):
# 帧数统计变量初始化
self.frameCounter = 0
def feedCap(self, im, func_status):
# 初始化dict,用于返回结果, 在这里初始化了键list_of_ids, 但后面没用到
retDict = {
'frame': None,
'list_of_ids': None,