search

ENCP / CNNdroid

Open Source Library for GPU-Accelerated Execution of Trained Deep Convolutional Neural Networks on Android
MIT License
539 stars 181 forks source link

Support MobileNet #21

Open john1deer opened 7 years ago

john1deer commented 7 years ago

MobileNet is a new architecture from Google (https://arxiv.org/abs/1704.04861) that is designed to run on mobile devices making it a perfect fit for cnndroid: https://github.com/Zehaos/MobileNet

I believe it's not supported due to 1x1 convolutions

changtimwu commented 7 years ago

MobileNet architecture heavily uses depthwise separable convolution. So the first step is implementing depthwiseConv with renderscript.

Reference: depthwise separable convolution in Metal render language http://machinethink.net/blog/googles-mobile-net-architecture-on-iphone/?utm_source=tuicool&utm_medium=referral