基於PyTorch的實用Library

本文紀錄一些基於PyTorch的深度學習Library

pprp/SimpleCVReproduction

pprp/awesome-attention-mechanism-in-cv

facebookresearch/detectron2

jacobgil/pytorch-grad-cam

xmu-xiaoma666/External-Attention-pytorch

from attention.SEAttention import SEAttention
import torch

input=torch.randn(50,512,7,7)
se = SEAttention(channel=512,reduction=8)
output=se(input)
print(output.shape)

thuml/Transfer-Learning-Library

facebookresearch/DomainBed

python3 -m domainbed.scripts.train\
--data_dir=./domainbed/data/MNIST/\
--algorithm IGA\
--dataset ColoredMNIST\
--test_env 2

TorchSSL/TorchSSL

adobe/antialiased-cnns

KevinMusgrave/pytorch-metric-learning

MadryLab/robustness

BorealisAI/advertorch

Machine Learning | Deep Learning | https://linktr.ee/yanwei