attention-module

Here are 5 public repositories matching this topic...

AhmadJeddi / Resnet-DCBAM

A clean and modular PyTorch implementation of DCBAM (Dual-Channel & Spatial Attention) integrated into a custom ResNet50 backbone. Includes attention modules, pretrained weight loading, layer freezing, full examples, and production-ready architecture.

machine-learning computer-vision deep-learning backbone cnn pytorch neural-networks resnet50 cbam attention-module dcbam

Updated Jul 26, 2026
Python

developer0hye / ZAM

Star

ZAM: Zero parameter Attention Module

attention bam self-attention cbam zam attention-module zcbam

Updated Oct 2, 2019
Python

rasoulameri / Defect_Segmentation

Star

A hybrid CNN–Transformer framework for precise industrial surface defect detection and segmentation, integrating Vision Transformer (ViT) with convolutional modules to effectively capture both local texture details and global contextual features.

python pytorch transformer vision segmentation convolutional-neural-networks industrial image-segmentation local-features global-features defect-detection attention-module surface-defect-detection vision-transformer transunet defect-segmentation cnn-transformer

Updated Oct 13, 2025
Python

zhaohao0404 / Focal-IoU-loss

Star

This code for solving the imbalance problems in regression head of object detectors. Comparison methods contain:Smooth-l1, IoU-loss, GIoU-loss, DIoU-loss, CIoU-loss.

pytorch attention-module iou-based-loss