Repository navigation

#

cbam

🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐

Python
11884
4 个月前

This is a pytorch repository of YOLOv4, attentive YOLOv4 and mobilenet YOLOv4 with PASCAL VOC and COCO

Python
1674
1 年前

YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quantization (MQBench) and Deployment (TensorRT, ncnn) Compression Tool Box.

Python
1006
3 年前

a pytorch lib with state-of-the-art architectures, pretrained models and real-time updated results

Python
863
4 年前

More readable and flexible yolov5 with more backbone(gcn, resnet, shufflenet, moblienet, efficientnet, hrnet, swin-transformer, etc) and (cbam,dcn and so on), and tensorrt

Python
673
8 个月前

Model Compression—YOLOv3 with multi lightweight backbones(ShuffleNetV2 HuaWei GhostNet), attention, prune and quantization

Python
497
4 年前

CBAM implementation on TensowFlow

Python
148
7 年前

[MICCAI'24] Official implementation of "BGF-YOLO: Enhanced YOLOv8 with Multiscale Attentional Feature Fusion for Brain Tumor Detection".

Python
111
3 个月前

CBAM implementation on TensorFlow Slim

Python
106
7 年前

CBAM(Convolutional Block Attention Module) implementation on TensowFlow2.0

Python
75
4 年前

可以训练yolov5(v6.0)、yolox、小型网络,添加注意力机制

Python
66
3 年前

This repo contains the 3D implementation of the commonly used attention mechanism for imaging.

Python
46
3 年前

CBAM: Convolutional Block Attention Module for CIFAR100 on VGG19

Jupyter Notebook
38
1 年前

Spatial and Channel Attention in CNN Architectures for Image Classification task

Jupyter Notebook
27
5 年前

ZAM: Zero parameter Attention Module

Python
26
6 年前

PHP package to read metadata and extract covers from eBooks, comics and audiobooks.

PHP
25
2 个月前

Hyperspectral Unmixing via Dual Attention Convolutional Neural Networks | 基于双注意力卷积神经网络的高光谱图像解混

Python
24
10 个月前

This is a torchvision style CNN models collection based on pytorch.

Python
21
5 年前

Code for paper "Channel Pruning Guided by Spatial and Channel Attention for DNNs in Intelligent Edge Computing"

Python
17
2 年前

pytorch implementation of several CNNs for image classification

Python
16
5 年前