Abstract:Computational resources are limited on real-time embedded devices, so the available computing cost of deployment on the target platform must be considered. We develop a feature extraction module based on the MobileNet backbone that can be adjusted in terms of computational complexity and capacity using the depth multiplier parameter, classifier depth, and kernel depth. These three parameters allow us to control the count of channels within the network, effectively managing the model's capacity and computational requirements. To achieve semantic segmentation, we incorporate additional components, such as an extension module. This extension module typically includes 1x1 pointwise convolutional layers for pixel-level classification and a transposed convolutional layer for upsampling the output to the original input image size. By combining the feature extraction module with this extension module, we can create a complete architecture capable of performing semantic segmentation tasks. The feature extraction module provides the initial feature extraction and the extension module adds the necessary components for accurate pixel-wise classification and upsampling. Compared to Hardware-aware Neural Architecture Search (NAS), pruning, runtime pruning, and knowledge distillation methods, our model has several advantages in terms of modular design, structural controllability, ease of implementation, and cost-effectiveness. Our computational efficiency, as measured by FLOPS, is highly competitive. Our method is distinguished by solving the problem of MobileNet's inability to adjust the size and number of convolution kernels. It achieves this through adaptable parameter tuning, including MobileNet's depth multiplier, the kernel size in the FCN head's Separable Convolution layer, and the depth of the first Point-wise Convolution layer. These adjustments are customized to match hardware's max multiply-accumulates (MACs), optimizing network capacity and maximizing resource utilization.

Mobile-Based Portrait Segmentation Network With Attention Mechanism

EHANet: Efficient Hybrid Attention Network Towards Real-time Semantic Segmentation

A Scalable Real-time Semantic Segmentation Network for Autonomous Driving

Edge Segmentation: Empowering Mobile Telemedicine with Compressed Cellular Neural Networks

Boundary-Aware Network for Fast and High-Accuracy Portrait Segmentation

ELA-Net: An Efficient Lightweight Attention Network for Skin Lesion Segmentation

PortraitNet: Real-time portrait segmentation network for mobile device

Lightweight Portrait Segmentation Via Edge-Optimized Attention

Lightweight Attention Network for Very High-Resolution Image Semantic Segmentation

MEDANet: More Efficient Dual Attention Network for Scene Segmentation

Attention based lightweight asymmetric network for real-time semantic segmentation

A Lightweight YOLOv5-Based Model with Feature Fusion and Dilation Convolution for Image Segmentation

M-FasterSeg: An efficient semantic segmentation network based on neural architecture search

DARSegNet: A Real-Time Semantic Segmentation Method Based on Dual Attention Fusion Module and Encoder-Decoder Network

Lw-TISNet: Light-Weight Convolutional Neural Network Incorporating Attention Mechanism and Multiple Supervision Strategy for Tongue Image Segmentation

When Humans Meet Machines: Towards Efficient Segmentation Networks.

DPANET:Dual Pooling Attention Network for Semantic Segmentation

LEDNet: A Lightweight Encoder-Decoder Network for Real-Time Semantic Segmentation

Threshold Attention Network for Semantic Segmentation of Remote Sensing Images

Feature Fusion Network Based on Hybrid Attention for Semantic Segmentation

MDNet: A Multi-image Input Real-Time Semantic Segmentation Model with Decoupled Supervision