vaibhav02kadam / Attention-U2Net

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Attention U2Net

Owing to the powerful feature extraction capabilities of convolutional neural networks (CNNs), deep learning methods have recently made significant strides in the field of salient object detection. Despite this progress, many current deep models struggle to effectively learn informative contextual features, resulting in suboptimal performance when faced with complex scenes. In our proposed approach, we explore the application of the U2Net-attention framework to enhance this performance of Salient Object Detection (SOD). Specifically, we analyze standard and modified attention blocks in conjunction with various U2Net architectures, such as attention-inside, attention-outside, and a combination of both. To provide a comprehensive evaluation of the proposed models, we employ a qualitative assessment of SOD using all attention-enhanced models and the attention maps generated.

Approach

This is the diagram that guides about our approach:

un2et-att

The following is the attention methods used: attent_block_combined

Results

result

Link to the Youtube video

About


Languages

Language:Python 100.0%