Skip to content

PyTorch and TensorFlow2 implementation of MobileNeXt

Notifications You must be signed in to change notification settings

romulus0914/MobileNeXt

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

MobileNeXt

A TensorFlow and PyTorch implementation of MobileNeXt

Overview

A TensorFlow and PyTorch implementation of MobileNeXt architecture: Rethinking Bottleneck Structure for Efficient Mobile Network Design. The authors rethink the necessity of inverted residual block and find it may bring risks of information loss and gradient confusion. They thus propose to flip the structure and present a novel bottleneck design, called the sandglass block, that performs identity mapping and spatial transformation at higher dimensions and thus alleviates information loss and gradient confusion effectively. In ImageNet classification, by simply replacing the inverted residual block with the sandglass block without increasing parameters and computation, the classification accuracy can be improved by more than 1.7% over MobileNetV2.

SandGlass Block

MobileNeXt v.s. MobileNetv2

Disclaimer

This is not the official implementation of MobileNeXt.

About

PyTorch and TensorFlow2 implementation of MobileNeXt

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages