GAIN

Object attention framework

A PyTorch implementation of an attention-guided inference network to focus on specific areas of objects in images

Pytorch implementation of paper: Tell Me Where to Look: Guided Attention Inference Network

GitHub

48 stars
2 watching
18 forks
Language: Python
last commit: about 5 years ago

Related projects:

Repository Description Stars
javeywang/pyramid-attention-networks-pytorch An implementation of a deep learning model using PyTorch for semantic segmentation tasks. 235
tengshaofeng/residualattentionnetwork-pytorch Implementing a deep learning framework for image classification using Residual Attention Network architecture 680
nexusapoorvacus/deepvariationstructuredrl An implementation of reinforcement learning for visual relationship and attribute detection using PyTorch. 63
kaiyangzhou/dassl.pytorch A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. 1,217
fgxaos/pytorch-innvestigate PyTorch implementation of an explainability technique for deep neural networks 9
achaiah/pywick A PyTorch-based neural network training framework with advanced features and utilities 398
jongchan/attention-module Provides implementations of attention modules for computer vision tasks using PyTorch 2,061
bearpaw/pytorch-classification An open-source project providing a unified interface for training and evaluating classification models using PyTorch. 1,692
wannabeog/mask-rcnn A PyTorch implementation of the Mask R-CNN architecture 992
pyg-team/pytorch-frame A deep learning framework for handling heterogeneous tabular data with diverse column types 543
hszhao/psanet A deep learning framework for semantic segmentation with spatial attention mechanisms 216
pp00704831/banet-tip-2022 A PyTorch implementation of an attention network for dynamic scene deblurring 37
orobix/visual-feature-attribution-using-wasserstein-gans-pytorch A PyTorch implementation of a feature attribution technique using Wasserstein Generative Adversarial Networks for anomaly detection in medical images. 93
benedekrozemberczki/gam An implementation of a graph classification model using structural attention and PyTorch 268
felixgwu/img_classification_pk_pytorch A PyTorch project for comparing image classification models and facilitating quick experiment setup 365