#

cbam

Here are 22 public repositories matching this topic...

xmu-xiaoma666 / External-Attention-pytorch

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

⭐

⭐

paper pytorch attention squeeze excitation-networks cbam linear-layers visual-tasks

Updated Jun 9, 2022
Python

argusswift / YOLOv4-pytorch

Open

训练自己的数据集的时候怎么放置啊？一定设置成voc2007与voc2012的形式吗？

13

haobabuhaoba commented Oct 6, 2020

[2020-10-06 21:13:34,984]-[train.py line:164]:===== Validate =====
Traceback (most recent call last):
  File "train.py", line 212, in <module>
    fp_16=opt.fp_16).train()
  File "train.py", line 166, in train
    APs, inference_time = Evaluator(self.yolov4, showatt=False).APs_voc()
  File "D:\hua\YOLOv4-pytorch\eval\evaluator.py", line 32, in APs_voc
    with open(img_inds_file, 'r')

Read more

good first issue

Open

Is it trained after loading the pre training model or not

16

implus / PytorchInsight

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

Updated Dec 8, 2020
Python

HaloTrouvaille / YOLO-Multi-Backbones-Attention

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

yolo attention-mechanism eca se shufflenet cbam huawei-ghostnet object-detection-datasets

Updated Nov 3, 2020
Python

Bobo-y / flexible-yolov5

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

backbone pytorch resnet object-detection se tensorrt neck shufflenet cbam gnn efficientnet hrnet dcnv2 yolov5 moblienet swin-transformer

Updated Jul 9, 2022
Python

Gumpest / YOLOv5-Multibackbone-Compression

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

tensorrt ncnn cbam network-slimming eagleeye ghostnet-yolov5 swin-transformer pplcnet tph-yolov5 shufflenetv2-yolov5 mobilenetv3small-yolov5 mqbench efficientnetlite-yolov5

Updated Apr 29, 2022
Python

kobiso / CBAM-tensorflow

CBAM implementation on TensowFlow

tensorflow resnext senet inception-resnet-v2 cbam inception-v4

Updated Aug 31, 2018
Python

kobiso / CBAM-tensorflow-slim

CBAM implementation on TensorFlow Slim

tensorflow slim resnet senet inception-resnet-v2 cbam inception-v4

Updated Sep 14, 2018
Python

zhangkaifang / CBAM-TensorFlow2.0

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

tensorflow tf2 attention resnet attention-mechanism tf cbam cbam-resnet

Updated Dec 23, 2020
Python

xialuxi / yolov5_all

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

mobilenet shufflenet senet cbam yolov5 yolox

Updated Nov 23, 2021
Python

Knight825 / models-pytorch

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

pytorch attention-mechanism cnn-model senet self-attention cbam cbam-resnet se-resnet crossstagepartial csp-resnet csp-densenet exactfusionmodel

Updated Sep 15, 2020
Python

developer0hye / ZAM

ZAM: Zero parameter Attention Module

attention bam self-attention cbam zam attention-module zcbam

Updated Oct 2, 2019
Python

doublechenching / xray14-keras

training a classification model with xray14 dataset

keras attention-model xception xray14 cbam

Updated Oct 1, 2018
Jupyter Notebook

syt2 / CNN

pytorch implementation of several CNNs for image classification

cnn pytorch imagenet image-classification resnet cifar resnext pretrained-weights senet cbam sknet train-from-scratch

Updated Jul 12, 2020
Python

arp95 / cbam_cnn_architectures_image_classification

Spatial and Channel Attention in CNN Architectures for Image Classification task

python pytorch image-classification cbam attention-cnn cnn-architectures

Updated Sep 12, 2020
Jupyter Notebook

laugh12321 / 3D-Attention-Keras

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

attention attention-model 3d-attention spatial-attention cbam channel-attention position-attention

Updated May 22, 2021
Python

zhangkaifang / ResNet

A minimal Tensorflow2.0 implementation of Resnet on CIFAR10 dataset.

tensorflow resnet attention-mechanism cifar10 csdn cifar100 cbam tensorflow2 cbam-resnet

Updated Mar 19, 2021
Python

x670783915 / huaweiyun_garbage_classify__learning

The topic was from huawei cloud garbage classification competition.

python classification pytroch cbam

Updated Dec 16, 2020
Python

firasl / inverse_attention

This repository provides the official implementation of 'Learning to ignore: rethinking attention in CNNs' accepted in BMVC 2021.

computer-vision attention attention-model attention-is-all-you-need cbam tensorflow2 learning-by-ignoring

Updated Mar 7, 2022
Python

fangvv / CPSCA

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

dnn vgg resnet attention-mechanism model-compression edge-computing senet channel-pruning cbam

Updated Nov 30, 2021
Python

DACN

laugh12321 / DACN

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

hyperspectral keras-tensorflow attention-model hyperspectral-unmixing cbam hyperspectral-images spectral-spatial dacn

Updated Jan 8, 2022
Python

haruishi43 / bam-cbam

Sponsor

PyTorch Code for BAM: Botteneck Attention Module and CBAM: Convolutional Block Attention Module

pytorch attention resnet bam cbam

Updated Sep 18, 2020
Python

Improve this page

Add a description, image, and links to the cbam topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the cbam topic, visit your repo's landing page and select "manage topics."