scrfd人脸检测测试

pip install mmdet 自动安装mmcv

win10上可以，服务器上没成功

报错：

KeyError: 'SCRFD is not in the models registry'

测试代码：

  

   

    

     

    
    

     

      import argparse
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import cv2
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      from mmdet.apis import inference_detector, init_detector
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def parse_args():
     
    
   

    

     

    
    

     

       parser = argparse.ArgumentParser(description='MMDetection webcam demo')
     
    
   

    

     

    
    

     

       parser.add_argument('--config',default="../configs/scrfd/scrfd_10g.py", help='test config file path')
     
    
   

    

     

    
    

     

       parser.add_argument('--checkpoint',default="model.pth",help='checkpoint file')
     
    
   

    

     

    
    

     

       parser.add_argument('--device', type=str, default='cpu', help='CPU/cuda:0 device option')
     
    
   

    

     

    
    

     

       parser.add_argument('--score-thr', type=float, default=0.5, help='bbox score threshold')
     
    
   

    

     

    
    

     

       args = parser.parse_ar
     
    
  
 

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

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