liuaoy

Image-recognition-of-agricultural-diseases

使用现代卷积神经网络架构（例如ResNet，DenseNet）对38类植物病害进行识别，并生成一个简单的UI操作界面

ASTGNN

This is a Pytorch implementation of ASTGNN. Now the corresponding paper is available online at https://ieeexplore.ieee.org/document/9346058.

AdjustAutocorrelation

Adjusting for Autocorrelated Errors in Neural Networks for Time Series

coatnet-pytorch

A PyTorch implementation of "CoAtNet: Marrying Convolution and Attention for All Data Sizes".

ConvNeXt

Code release for ConvNeXt model

deep-learning-for-image-processing

deep learning for image processing including classification and object-detection etc.

deep-time-series

Deep learning PyTorch library for time series forecasting

deit

Official DeiT repository

External-Attention-pytorch

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

FFVT

The official PyTorch implementation of Feature Fusion Vision Transformer for Fine-Grained Visual Categorization (BMVC 2021).

focal_loss_pytorch

A PyTorch Implementation of Focal Loss.

forlove

GAT_traffic_prediction

基于图注意力模型（GAT）的交通网络流量预测

Graph-WaveNet

Modifications to Graph Wavenet

LeafGAN

open-aff

code and trained models for "Attentional Feature Fusion"

pretty-print-confusion-matrix

Confusion Matrix in Python: plot a pretty confusion matrix (like Matlab) in python using seaborn and matplotlib

pytorch-cnn-visualizations

Pytorch implementation of convolutional neural network visualization techniques

pytorch-grad-cam

Many Class Activation Map methods implemented in Pytorch for CNNs and Vision Transformers. Including Grad-CAM, Grad-CAM++, Score-CAM, Ablation-CAM and XGrad-CAM

pytorch-loss

label-smooth, amsoftmax, focal-loss, triplet-loss, lovasz-softmax. Maybe useful

pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers

Ranger-Deep-Learning-Optimizer

Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase

Ranger21

Ranger deep learning optimizer rewrite to use newest components

SCINet

The GitHub repository for the paper: “Time Series is a Special Sequence: Forecasting with Sample Convolution and Interaction“.

STEP

Code for our SIGKDD'22 paper Pre-training-Enhanced Spatial-Temporal Graph Neural Network For Multivariate Time Series Forecasting.

STNN

STNN-ICDM21

Swin-Transformer

Implementation of Swin Transformer with Pytorch

Trafficformer

Trafficformer: Transformer based traffic prediction model, code of paper

vision

Datasets, Transforms and Models specific to Computer Vision

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch