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]