LFD-A-Light-and-Fast-Detector

开源地址

https://github.com/YonghaoHe/LFD-A-Light-and-Fast-Detector

320*240 cpu 17 90ms，模型大小 6.71 m

LFD-A-Light-and-Fast-Detector

  

   

    

     

    
    

     

      #!/usr/bin/env python3
     
    
   

    

     

    
    

     

      # -*- coding: utf-8 -*-
     
    
   

    

     

    
    

     

      """
     
    
   

    

     

    
    

     

      Created on Tue May 21 09:09:25 2019
     
    
   

    

     

    
    

     

      Pytorch mobilefacenet & arcface architecture
     
    
   

    

     

    
    

     

      
     
    
   

    

     

    
    

     

      @author: AIRocker
     
    
   

    

     

    
    

     

      """
     
    
   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      from torch.nn import Linear, Conv2d, BatchNorm1d, BatchNorm2d, PReLU, ReLU, Sigmoid, Dropout2d, Dropout, AvgPool2d, MaxPool2d, AdaptiveAvgPool2d, Sequential, Module, Parameter
     
    
   

    

     

    
    

     

      from torch import nn
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      import math
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      from cv_tools.call_cmd import onnx2mnn
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class Flatten(Module):
     
    
   

    

     

    
    

     
 def forward(self, input):
     
    
   

    

     

    
    

     
 return input.view(input.size(0), -1)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def l2_norm(input,axis=1):
     
    
   

    

     

    
    

     

       norm = torch.norm(input,2,axis,True)
     
    
   

    

     

    
    

     

       output = torch.div(input, norm)
     
    
   

    

     

    
    

     
 if torch.isnan(output[0]
     
    
  
 

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

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