ShiTengfei's starred repositories
TransferAttack
TransferAttack is a pytorch framework to boost the adversarial transferability for image classification.
Torch-Pruning
[CVPR 2023] Towards Any Structural Pruning; LLMs / SAM / Diffusion / Transformers / YOLOv8 / CNNs
Beyond-ImageNet-Attack
Beyond imagenet attack (accepted by ICLR 2022) towards crafting adversarial examples for black-box domains.
Neural-Network-Parameter-Diffusion
We introduce a novel approach for parameter generation, named neural network parameter diffusion (p-diff), which employs a standard latent diffusion model to synthesize a new set of parameters
pytorch-AdaIN
Unofficial pytorch implementation of 'Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization' [Huang+, ICCV2017]
Stochastic-Gradient-Aggregation
Official implementation of the ICCV2023 paper: Enhancing Generalization of Universal Adversarial Perturbation through Gradient Aggregation
segment-anything
The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.
annotated_deep_learning_paper_implementations
🧑🏫 60 Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠
PyTorchRelevancePropagation
A basic implementation of Layer-wise Relevance Propagation (LRP) in PyTorch.
pytorch-grad-cam
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
pytorch-adversarial-attack
Implementation of gradient-based adversarial attack(FGSM,MI-FGSM,PGD)
Generative_Adversarial_Perturbations
Generative Adversarial Perturbations (CVPR 2018)
StyleTransfer
Implementation of "Perceptual Losses for Real-Time Style Transfer and Super-Resolution" in PyTorch
conv_arithmetic
A technical report on convolution arithmetic in the context of deep learning
Self-Universality
Enhancing the Self-Universality for Transferable Targeted Attacks [CVPR 2023 Paper]
pytorch-image-models
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more