yolov5-face无关键点测试

本文只是把关键点部分去掉了，网络部分还没改。

可以用在别的物体检测领域,测试ok

  
  

   
   

    
    

     
     

    
    
    
    

     
     

      # -*- coding: UTF-8 -*-
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      import argparse
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      import time
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      from pathlib import Path
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     
 
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      import cv2
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      import torch
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      import torch.backends.cudnn as cudnn
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      from numpy import random
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      import copy
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     
 
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      from models.experimental import attempt_load
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      from utils.datasets import letterbox
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      from utils.general import check_img_size, non_max_suppression_face, apply_classifier, scale_coords, xyxy2xywh, \
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      
       strip_optimizer, set_logging, increment_path
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      from utils.plots import plot_one_box
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      from utils.torch_utils import select_device, load_classifier, time_synchronized
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     
 
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     
 
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      def load_model(weights, device):
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      
       model = attempt_load(weights, map_location=device)  # load FP32 model
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     
 return model
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     
 
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     
 
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      def scale_coords_landmarks(img1_shape, coords, img0_shape, ratio_pad=None):
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     
 # Rescale coords (xyxy) from img1_shape to img0_shape
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     
 if ratio_pad is None:  # calculate from img0_shape
     
     
    
    
   
   

    
    

     
     

    
    
    
    

     
     

      
       gain = min(img1_shape[
     
     
    
    
  
  
 
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/118297123