novice03 / mobiledet-pytorch

PyTorch Implementation of MobileDet (https://arxiv.org/abs/2004.14525v3) backbones.
11 stars 6 forks source link
object-detection pytorch

MobileDet Backbones

PyTorch implementation of MobileDet backbones introduced in MobileDets: Searching for Object Detection Architectures for Mobile Accelerators. Ross Wightman's timm library has been used for some helper functions and inspiration for syntax style. The following are the main blocks used in the tensorflow implementation and their corresponding blocks in the timm library:

Tensorflow timm
fused_conv EdgeResidual
inverterted_bottleneck InvertedResidual
inverted_bottleneck_no_expansion DepthwiseSeparableConv

tucker-conv package is used for tucker_conv