yolov5 face 获取漏检图片

最新推荐文章于 2024-10-03 09:02:12 发布

AI算法网奇

最新推荐文章于 2024-10-03 09:02:12 发布

阅读量420

点赞数

分类专栏：深度学习宝典文章标签： python pytorch 开发语言

本文链接：https://blog.csdn.net/jacke121/article/details/131148129

版权

深度学习宝典专栏收录该内容

989 篇文章 714 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了如何使用 YOLOv5 Face 模型在目标检测过程中捕获漏检的人脸图片，通过Python和PyTorch进行深度学习应用，详细阐述了实现步骤和关键代码。

摘要由CSDN通过智能技术生成

yolov5 face 获取漏检图片

# -*- coding: UTF-8 -*-
import argparse
import time
from pathlib import Path
import sys
import os

import numpy as np
import cv2
import torch
import torch.backends.cudnn as cudnn
from numpy import random
import copy

from torch import nn

from common import Conv, DWConv
from tools.datasets import letterbox, img_formats, vid_formats, LoadImages, LoadStreams
from tools.general import check_img_size, non_max_suppression_face, apply_classifier, scale_coords, xyxy2xywh, strip_optimizer, set_logging, increment_path

class Ensemble(nn.ModuleList):
    # Ensemble of models
    def __init__(self):
        super(Ensemble, self).__init__()

    def forward(self, x, augment=False):
        y = []
        for module in self:

了解本专栏